We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4f8ca40 commit ee032acCopy full SHA for ee032ac
examples/openvino/aot_openvino_compiler.py
@@ -187,7 +187,8 @@ def main(
187
batch_size = calibration_dataset.batch_size
188
subset_size = (subset_size // batch_size) + int(subset_size % batch_size > 0)
189
190
- transform_fn = lambda x: x[0]
+ def transform_fn(x):
191
+ return x[0]
192
quantized_model = quantize_model(
193
aten_dialect.module(),
194
calibration_dataset,
0 commit comments