Skip to content

T
tic
Commit bbc78221 by lixiaoquan Committed by Lianmin Zheng
Options

[FRONTEND][TENSORFLOW] Support AttrValue that has different types of value in a list (#2177)

parent acea3cc3
Showing with 2 additions and 2 deletions
nnvm/python/nnvm/frontend/tensorflow.py
...@@ -1280,9 +1280,9 @@ class GraphProto(object): ...@@ -1280,9 +1280,9 @@ class GraphProto(object):
for f in fields: for f in fields:
if getattr(x.list, f): if getattr(x.list, f):
if f == "type": if f == "type":
ret = [dtypes.as_dtype(x) for x in list(getattr(x.list, f))] ret += [dtypes.as_dtype(x) for x in list(getattr(x.list, f))]
else: else:
ret = list(getattr(x.list, f)) ret += list(getattr(x.list, f))
else: else:
for f in fields: for f in fields:
if x.HasField(f): if x.HasField(f):
......
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