fix lint

Matthew · Matthew · commit 1bf48a410a89 · 2021-08-30T10:49:01.000-06:00
diff --git a/python/tvm/relay/transform/fake_quantization_to_integer.py b/python/tvm/relay/transform/fake_quantization_to_integer.py
@@ -139,9 +139,8 @@ def conv2d(expr, type_map):
     out = relay.qnn.op.conv2d(
         x, weight, x_t.zero_point, w_t.zero_point, x_t.scale, w_t.scale, **attrs
     )
-    scale_shape = infer_shape(conv_scale)
     out_layout = attrs["out_layout"] if attrs["out_layout"] != "" else attrs["data_layout"]
-    out_axis = tvm.tir.bijective_layout(out_layout, "NCHW").backward_index(list(range(4)))[1]
+    out_axis = bijective_layout(out_layout, "NCHW").backward_index(list(range(4)))[1]
     return [out, TensorAffineType(conv_scale, conv_zp, out.attrs.out_dtype, out_axis.value)]
 
 

Original file line number	Diff line number	Diff line change
`@@ -139,9 +139,8 @@ def conv2d(expr, type_map):`
`139`	`139`	`out = relay.qnn.op.conv2d(`
`140`	`140`	`x, weight, x_t.zero_point, w_t.zero_point, x_t.scale, w_t.scale, **attrs`
`141`	`141`	`)`
`142`		`- scale_shape = infer_shape(conv_scale)`
`143`	`142`	`out_layout = attrs["out_layout"] if attrs["out_layout"] != "" else attrs["data_layout"]`
`144`		`- out_axis = tvm.tir.bijective_layout(out_layout, "NCHW").backward_index(list(range(4)))[1]`
	`143`	`+ out_axis = bijective_layout(out_layout, "NCHW").backward_index(list(range(4)))[1]`
`145`	`144`	`return [out, TensorAffineType(conv_scale, conv_zp, out.attrs.out_dtype, out_axis.value)]`
`146`	`145`
`147`	`146`