Commit bbc78221 by lixiaoquan Committed by Lianmin Zheng

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

parent acea3cc3
......@@ -1280,9 +1280,9 @@ class GraphProto(object):
for f in fields:
if getattr(x.list, f):
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:
ret = list(getattr(x.list, f))
ret += list(getattr(x.list, f))
else:
for f in fields:
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