saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use == instead of is for python 3.8

Browse source
This commit is contained in:
Yufeng Li 2021-02-09 15:56:05 -08:00 committed by Changming Sun
parent e70344e648
commit 505c1f30b5
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -42,7 +42,7 @@ class QDQQuantizer(ONNXQuantizer):
self.tensors_to_quantize.append(tensor_name)
elif tensor_name in self.value_infos.keys():
vi = self.value_infos[tensor_name]
if vi.type.HasField('tensor_type') and vi.type.tensor_type.elem_type is TensorProto.FLOAT:
if vi.type.HasField('tensor_type') and vi.type.tensor_type.elem_type == TensorProto.FLOAT:
self.tensors_to_quantize.append(tensor_name)
else:
logging.warning(