From 6e6ac97924c238e367617a0036b71b0befb8a5a4 Mon Sep 17 00:00:00 2001 From: SRK Reddy Date: Thu, 6 Dec 2018 19:11:51 +0530 Subject: [PATCH] [FRONTEND][TENSORFLOW] Bugfix --- nnvm/python/nnvm/frontend/tensorflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nnvm/python/nnvm/frontend/tensorflow.py b/nnvm/python/nnvm/frontend/tensorflow.py index 26e59dc7e830..2a398c3f9635 100644 --- a/nnvm/python/nnvm/frontend/tensorflow.py +++ b/nnvm/python/nnvm/frontend/tensorflow.py @@ -1109,8 +1109,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.