@@ -105,23 +105,23 @@ def convert_variables_to_constants_large_model(func):
105
105
except ImportError :
106
106
_not_implemented_tf_placeholder ("_convert_variables_to_constants_v2_impl" )()
107
107
frozen_graph_def , _ = \
108
- _convert_variables_to_constants_v2_impl (func , lower_control_flow = False , aggressive_inlining = False )
108
+ _convert_variables_to_constants_v2_impl (func , lower_control_flow = False , aggressive_inlining = True )
109
109
return frozen_graph_def
110
110
111
111
try :
112
112
from tensorflow .python .framework .convert_to_constants import \
113
113
_FunctionConverterData , _replace_variables_by_constants # pylint: disable=protected-access
114
114
except ImportError :
115
115
_not_implemented_tf_placeholder ("_replace_variables_by_constants" )()
116
- converter_data = _FunctionConverterData (func = func , lower_control_flow = False , aggressive_inlining = False )
116
+ converter_data = _FunctionConverterData (func = func , lower_control_flow = False , aggressive_inlining = True )
117
117
frozen_graph_def , _ = _replace_variables_by_constants (converter_data = converter_data )
118
118
return frozen_graph_def
119
119
120
120
def from_function (func , input_names , output_names , large_model = False ):
121
121
if large_model :
122
122
return convert_variables_to_constants_large_model (func )
123
123
124
- frozen_func = convert_variables_to_constants_v2 (func , lower_control_flow = False )
124
+ frozen_func = convert_variables_to_constants_v2 (func , lower_control_flow = False , aggressive_inlining = True )
125
125
graph_def = frozen_func .graph .as_graph_def (add_shapes = True )
126
126
# output_names = [i.name for i in frozen_func.outputs]
127
127
tf_reset_default_graph ()
0 commit comments