Fix the regression that raised by matmul min_first feature.
@@ -236,8 +236,11 @@ class GraphConverter:
input_graph (tf.compat.v1.GraphDef): input graph
"""
graph = self.load_graph(input_graph)
+ if len(self.inputs) > 1:
input_tensor = [
graph.get_tensor_by_name(x + ":0") for x in self.inputs]
+ else:
+ input_tensor = graph.get_tensor_by_name(self.inputs[0] + ":0")
output_tensor = [
graph.get_tensor_by_name(x + ":0") for x in self.outputs
