saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-26 22:35:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix typo OpTypesToExcludeOutputQuantizatioin (#13096)

Browse source 
Change all occurences of `OpTypesToExcludeOutputQuantizatioin` into `OpTypesToExcludeOutputQuantization`
This commit is contained in:
fxmarty 2022-10-14 23:11:37 +02:00 committed by GitHub
parent c4a52820a5
commit 4fe6b23699
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/python/tools/quantization/qdq_quantizer.py
View file

@ -86,8 +86,8 @@ class QDQQuantizer(ONNXQuantizer):
# So, we don't recommend to add QDQ to node's output under such condition.
self.op_types_to_exclude_output_quantization = (
[]
if "OpTypesToExcludeOutputQuantizatioin" not in extra_options
else extra_options["OpTypesToExcludeOutputQuantizatioin"]
if "OpTypesToExcludeOutputQuantization" not in extra_options
else extra_options["OpTypesToExcludeOutputQuantization"]
)
# We do quantization on Dequantizelinear's input to remove Quantizelinear for weight as an optimization.

4
onnxruntime/python/tools/quantization/quantize.py
View file

@ -118,7 +118,7 @@ class StaticQuantConfig(QuantConfig):
Default is False which quantizes floating-point weight and feeds it to solely inserted
DeQuantizeLinear node. If True, it remains floating-point weight and inserts both
QuantizeLinear/DeQuantizeLinear nodes to weight.
OpTypesToExcludeOutputQuantizatioin = list of op type :
OpTypesToExcludeOutputQuantization = list of op type :
Default is []. If any op type is specified, it won't quantize the output of ops with this
specific op types.
DedicatedQDQPair = True/False :
@ -304,7 +304,7 @@ def quantize_static(
Default is False which quantizes floating-point weight and feeds it to solely inserted
DeQuantizeLinear node. If True, it remains floating-point weight and inserts both
QuantizeLinear/DeQuantizeLinear nodes to weight.
OpTypesToExcludeOutputQuantizatioin = list of op type :
OpTypesToExcludeOutputQuantization = list of op type :
Default is []. If any op type is specified, it won't quantize the output of ops with this
specific op types.
DedicatedQDQPair = True/False :

4
onnxruntime/test/python/quantization/test_qdq.py
View file

@ -101,7 +101,7 @@ class TestQDQExtraOptions(unittest.TestCase):
{
"ActivationSymmetric": True,
"AddQDQPairToWeight": True,
"OpTypesToExcludeOutputQuantizatioin": [],
"OpTypesToExcludeOutputQuantization": [],
},
) # extra_options
quantizer.quantize_model()
@ -202,7 +202,7 @@ class TestQDQExtraOptions(unittest.TestCase):
{
"ActivationSymmetric": True,
"AddQDQPairToWeight": True,
"OpTypesToExcludeOutputQuantizatioin": op_types_to_quantize,
"OpTypesToExcludeOutputQuantization": op_types_to_quantize,
"DedicatedQDQPair": True,
},
) # extra_options