diff --git a/model_compression_toolkit/metadata.py b/model_compression_toolkit/metadata.py index c200d2036..122b6ec18 100644 --- a/model_compression_toolkit/metadata.py +++ b/model_compression_toolkit/metadata.py @@ -57,6 +57,7 @@ class TPCVersions: tpc_minor_version: str = f'{fqc.tpc.tpc_minor_version}' tpc_patch_version: str = f'{fqc.tpc.tpc_patch_version}' tpc_platform_type: str = f'{fqc.tpc.tpc_platform_type}' + tpc_name: str = f'{fqc.tpc.name}' tpc_schema: str = f'{fqc.tpc.SCHEMA_VERSION}' return asdict(TPCVersions(mct_version))