Skip to content

Commit f8f998c

Browse files
committed
Fix node identity check
1 parent 0fef04a commit f8f998c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/xnnpack/_passes/channels_last_tagged_reshape_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,7 @@ def input_to_nhwc(
310310
if is_dequant:
311311
# Replace downstream input_nodes with NHWC node
312312
for user in list(input_node.users):
313-
if user != input_node_nhwc:
313+
if user is not input_node_nhwc:
314314
user.replace_input_with(input_node, input_node_nhwc)
315315
graph_module.recompile()
316316

0 commit comments

Comments
 (0)