Skip to content

Commit ffb5c63

Browse files
committed
pep8
1 parent e3bc51c commit ffb5c63

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tf2onnx/onnx_opset/quantize.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,10 @@
1313
import sys
1414

1515
import numpy as np
16-
from onnx import onnx_pb, numpy_helper, TensorProto
1716
from onnx.onnx_pb import TensorProto
1817

19-
from tf2onnx import constants, utils
20-
from tf2onnx.graph_builder import GraphBuilder
18+
from tf2onnx import utils
2119
from tf2onnx.handler import tf_op
22-
from tf2onnx.onnx_opset import nn, math
2320

2421
logger = logging.getLogger(__name__)
2522

@@ -71,7 +68,7 @@ def version_11(cls, ctx, node, **kwargs):
7168
shape = ctx.get_shape(node.input[0])
7269
axis = 1
7370
idtype = TensorProto.UINT8
74-
71+
7572
pb_scale = ctx.make_const(
7673
utils.make_name("{}_scaley".format(node.name)),
7774
np.array(scale, dtype=np.float32))
@@ -94,4 +91,3 @@ def version_11(cls, ctx, node, **kwargs):
9491
shapes=[shape], dtypes=[dtype])
9592
ctx.replace_all_inputs(ctx.get_nodes(), node.output[0], last_node.output[0])
9693

97-

0 commit comments

Comments
 (0)