Skip to content

Commit 919babf

Browse files
Apply suggestions from code review
Co-authored-by: TomWildenhain-Microsoft <[email protected]>
1 parent a4c8bbe commit 919babf

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
@@ -113,7 +113,7 @@ def _fold_transpose(node, graph) -> list:
113113
@staticmethod
114114
@_register_func("Reshape")
115115
def _fold_reshape(node, graph):
116-
const_val_data = node.inputs[0].get_tensor_value(as_list=False)
116+
const_val_data = node.inputs[0].get_tensor_value(as_list=False)
117117
const_val_shape = node.inputs[1].get_tensor_value(as_list=False)
118118
const_val_after_trans = const_val_data.reshape(const_val_shape)
119119
return [const_val_after_trans]

0 commit comments

Comments
 (0)