Skip to content

Commit 0adae73

Browse files
committed
Update const_fold_optimizer.py
1 parent eb68bcd commit 0adae73

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/optimizer/const_fold_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def _replace_node_with_const(node, graph, vals):
9090
const_node = graph.make_const(utils.make_name("const_fold_opt"), val)
9191
graph.set_dtype(const_node.output[0], utils.map_numpy_to_onnx_dtype(val.dtype))
9292
graph.set_shape(const_node.output[0], val.shape)
93-
graph.replace_all_inputs(graph.get_nodes(), old_input, const_node.output[0])
93+
graph.replace_all_inputs(None, old_input, const_node.output[0]) # graph.get_nodes()
9494
graph.remove_node(node.name)
9595

9696
@staticmethod

0 commit comments

Comments
 (0)