[FRONTEND][TENSORFLOW] Bugfix (apache#2267)

neo-ai · Feb 20, 2019 · 3f26d73 · 3f26d73
1 parent b6baff3
commit 3f26d73
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nnvm/python/nnvm/frontend/tensorflow.py b/nnvm/python/nnvm/frontend/tensorflow.py
@@ -1141,8 +1141,8 @@ def from_tensorflow(self, graph, layout="NHWC", shape=None, outputs=None):
                         tensor_value.tensor_shape)]
             elif '_output_shapes' in attr:
                 self._output_shapes[node.name] = \
-                    [tensor_util.TensorShapeProtoToList(shape) \
-                    for shape in attr['_output_shapes']]
+                    [tensor_util.TensorShapeProtoToList(tshape) \
+                    for tshape in attr['_output_shapes']]
             elif shape:
                 # Keep the list indexable to avoid key error.
                 # Actual value will be filled after node creation.