KPL Sequence template using ragged input.

PiperOrigin-RevId: 305822626
Change-Id: I5aaa4b506e39a403f176da3772dcd1338362686a
This commit is contained in:
Zhenyu Tan 2020-04-09 21:45:43 -07:00 committed by TensorFlower Gardener
parent f22174826d
commit 4f3652605a

View File

@ -178,7 +178,7 @@ def get_tensor_from_tensor_info(tensor_info, graph=None, import_scope=None):
spec = struct_coder.decode_proto(spec_proto)
components = [_get_tensor(component.name) for component in
tensor_info.composite_tensor.components]
return spec.from_components(components)
return spec._from_components(components) # pylint: disable=protected-access
else:
raise ValueError("Invalid TensorInfo.encoding: %s" % encoding)