From 505c1f30b59f662064bb252cd81aa8719a5c7b90 Mon Sep 17 00:00:00 2001 From: Yufeng Li Date: Tue, 9 Feb 2021 15:56:05 -0800 Subject: [PATCH] use == instead of is for python 3.8 --- onnxruntime/python/tools/quantization/qdq_quantizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/quantization/qdq_quantizer.py b/onnxruntime/python/tools/quantization/qdq_quantizer.py index 935b73ea65..a127fcaf46 100644 --- a/onnxruntime/python/tools/quantization/qdq_quantizer.py +++ b/onnxruntime/python/tools/quantization/qdq_quantizer.py @@ -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(