Skip to content

Commit 5214e20

Browse files
authored
Fix an option name in QDQQuantizer (#187)
1 parent e73b0e8 commit 5214e20

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: quantization/nlp/bert/trt/e2e_tensorrt_bert_example.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ def get_op_nodes_not_followed_by_specific_op(model, op1, op2):
274274
[], #nodes_to_quantize
275275
nodes_to_exclude,
276276
op_types_to_quantize,
277-
{'ActivationSymmetric' : True, 'AddQDQPairToWeight' : True, 'OpTypesToExcludeOutputQuantizatioin': op_types_to_quantize, 'DedicatedQDQPair': True, 'QDQOpTypePerChannelSupportToAxis': {'MatMul': 1} }) #extra_options
277+
{'ActivationSymmetric' : True, 'AddQDQPairToWeight' : True, 'OpTypesToExcludeOutputQuantization': op_types_to_quantize, 'DedicatedQDQPair': True, 'QDQOpTypePerChannelSupportToAxis': {'MatMul': 1} }) #extra_options
278278
quantizer.quantize_model()
279279
quantizer.model.save_model_to_file(qdq_model_path, False)
280280
print("QDQ model is saved to ", qdq_model_path)

0 commit comments

Comments
 (0)