Skip to content

Commit e2417b8

Browse files
author
Ofir Gordon
committed
Add a flag for TPC schema v2 for enabling quantization preserving quantizers in the output model
1 parent 7a6da34 commit e2417b8

File tree

1 file changed

+3
-0
lines changed
  • model_compression_toolkit/target_platform_capabilities/schema

1 file changed

+3
-0
lines changed

model_compression_toolkit/target_platform_capabilities/schema/v2.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,7 @@ class TargetPlatformCapabilities(BaseModel):
214214
add_metadata (bool): Flag to determine if metadata should be added.
215215
name (str): Name of the Target Platform Model.
216216
is_simd_padding (bool): Indicates if SIMD padding is applied.
217+
insert_preserving_quantizers (bool): Whether to include quantizers for quantization preserving operations in the quantized model.
217218
SCHEMA_VERSION (int): Version of the schema for the Target Platform Model.
218219
"""
219220
default_qco: QuantizationConfigOptions
@@ -224,7 +225,9 @@ class TargetPlatformCapabilities(BaseModel):
224225
tpc_platform_type: Optional[str]
225226
add_metadata: bool = True
226227
name: Optional[str] = "default_tpc"
228+
227229
is_simd_padding: bool = False
230+
insert_preserving_quantizers: bool = False
228231

229232
SCHEMA_VERSION: int = 2
230233

0 commit comments

Comments
 (0)