Searched refs:keys_to_tensors (Results 1 – 2 of 2) sorted by relevance
68 def tensors_to_item(self, keys_to_tensors): argument98 def tensors_to_item(self, keys_to_tensors): argument99 return self._func(keys_to_tensors)127 def tensors_to_item(self, keys_to_tensors): argument139 side = keys_to_tensors[key]190 def tensors_to_item(self, keys_to_tensors): argument191 tensor = keys_to_tensors[self._tensor_key]196 shape_dim = keys_to_tensors[k]242 def tensors_to_item(self, keys_to_tensors): argument243 unmapped_tensor = super(LookupTensor, self).tensors_to_item(keys_to_tensors)[all …]
671 def HandleDepth(keys_to_tensors): argument672 depth = list(keys_to_tensors.values())[0]697 def ConditionalDecoding(keys_to_tensors): argument699 image_buffer = keys_to_tensors['image/encoded']700 image_format = keys_to_tensors['image/format']