Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
inisis committed Dec 11, 2024
1 parent da66046 commit de34d61
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion onnxslim/core/optimization/dead_node_elimination.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ def dead_node_elimination(graph, is_subgraph=False):
isinstance(node.inputs[0], Constant) and isinstance(node.inputs[1], Variable)
):
idx, constant_variable = get_constant_variable(node, return_idx=True)
if np.all(constant_variable.values == 1):git
if np.all(constant_variable.values == 1):
var_idx = 0 if idx == 1 else 1
node.replace_all_uses_with(node, var_idx)
logger.debug(f"removing {node.op} op: {node.name}")
Expand Down

0 comments on commit de34d61

Please sign in to comment.