Skip to content

Commit b43f6d0

Browse files
committed
Update quantize.py
1 parent bcde434 commit b43f6d0

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tf2onnx/onnx_opset/quantize.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from __future__ import unicode_literals
1111

1212
import logging
13-
import sys
1413

1514
import numpy as np
1615
from onnx.onnx_pb import TensorProto
@@ -28,7 +27,7 @@
2827
class FakeQuantWithMinMaxArgs:
2928
# see https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/fake-quant-with-min-max-args
3029
@classmethod
31-
def version_11(cls, ctx, node, **kwargs):
30+
def version_10(cls, ctx, node, **kwargs):
3231
# hack to make up for the missing onnx pack op
3332
amin = node.get_attr("min").f
3433
amax = node.get_attr("max").f

0 commit comments

Comments
 (0)