Resolve the auto input/output detection issue raised by merge.
@@ -29,13 +29,12 @@ class PreOptimization(object):
self._tmp_graph_def = None
# self.tf_version = tf.version.VERSION
self._excluded_node_names = []
+ if not self.inputs or not self.outputs:
+ self.inputs, self.outputs = self.analyzer.get_graph_input_output()
def get_excluded_node_names(self):
return self._excluded_node_names
- if not self.inputs or not self.outputs:
- self.inputs, self.outputs = self.analyzer.get_graph_input_output()
-
def get_optimized_graphdef(self):
"""Executed the non-precision dependant graph optimization.
The input graph will be optimized with following passes:
