experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bug fix for composite tensors in SavedModel

Browse Source 
PiperOrigin-RevId: 311124917
Change-Id: I1e0b2fb6484cb9049445943594a1dec5760e2620
This commit is contained in:
Edward Loper 2020-05-12 07:56:12 -07:00 committed by TensorFlower Gardener
parent a672463c59
commit 773813a426
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/saved_model/utils_impl.py
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) # pylint: disable=protected-access
return nest.pack_sequence_as(spec, components, expand_composites=True)
else:
raise ValueError("Invalid TensorInfo.encoding: %s" % encoding)