Skip to content

Commit b44a0e4

Browse files
committed
Fix type checker issues
Signed-off-by: Per Åstrand <[email protected]> Change-Id: Iee35eeb4af28a037848570b1d5143380222f549d
1 parent acfd652 commit b44a0e4

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

backends/arm/_passes/fold_qdq_with_annotated_qparams_pass.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,7 @@ def call(self, graph_module: GraphModule) -> PassResult:
9999
for i, arg in enumerate(n.args):
100100
if not isinstance(arg, Node):
101101
continue
102+
arg = cast(Node, arg)
102103

103104
# Make sure arg has requires_grad set to False
104105
# For parameters that are not quantized, sometimes (i.e. convolution)

backends/arm/operators/op_max.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ def define_node(
4040
) -> None:
4141
assert inputs[0].dtype == inputs[1].dtype
4242

43+
scale_back = 1.0
4344
max_output = output
4445
if inputs[0].dtype == ts.DType.INT8:
4546
input_qparams = get_input_qparams(node)

backends/arm/operators/op_min.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ def define_node(
4141
) -> None:
4242
assert inputs[0].dtype == inputs[1].dtype
4343

44+
scale_back = 1.0
4445
min_output = output
4546
if inputs[0].dtype == ts.DType.INT8:
4647
input_qparams = get_input_qparams(node)

0 commit comments

Comments
 (0)