Skip to content

T
tic
Commit df9d3ad2 by Siva Committed by Tianqi Chen
Options

[FRONTEND][TENSORFLOW] Bugfix (#2267)

parent aa1310d1
Showing with 2 additions and 2 deletions
nnvm/python/nnvm/frontend/tensorflow.py
...@@ -1124,8 +1124,8 @@ class GraphProto(object): ...@@ -1124,8 +1124,8 @@ class GraphProto(object):
tensor_value.tensor_shape)] tensor_value.tensor_shape)]
elif '_output_shapes' in attr: elif '_output_shapes' in attr:
self._output_shapes[node.name] = \ self._output_shapes[node.name] = \
[tensor_util.TensorShapeProtoToList(shape) \ [tensor_util.TensorShapeProtoToList(tshape) \
for shape in attr['_output_shapes']] for tshape in attr['_output_shapes']]
elif shape: elif shape:
# Keep the list indexable to avoid key error. # Keep the list indexable to avoid key error.
# Actual value will be filled after node creation. # Actual value will be filled after node creation.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment