Skip to content

Commit 110ddec

Browse files
Disable aggressive_inlining before tf 2.2
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent a6f370e commit 110ddec

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tf2onnx/tf_loader.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,10 @@ def from_function(func, input_names, output_names, large_model=False):
121121
if large_model:
122122
return convert_variables_to_constants_large_model(func)
123123

124-
frozen_func = convert_variables_to_constants_v2(func, lower_control_flow=False, aggressive_inlining=True)
124+
if get_tf_version() < LooseVersion("2.2"):
125+
frozen_func = convert_variables_to_constants_v2(func, lower_control_flow=False)
126+
else:
127+
frozen_func = convert_variables_to_constants_v2(func, lower_control_flow=False, aggressive_inlining=True)
125128
graph_def = frozen_func.graph.as_graph_def(add_shapes=True)
126129
# output_names = [i.name for i in frozen_func.outputs]
127130
tf_reset_default_graph()

0 commit comments

Comments
 (0)