Searched refs:flat_value (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 137 def _from_tensor_list(self, flat_value): argument
152 def _from_compatible_tensor_list(self, flat_value): argument
167 return self._from_tensor_list(flat_value)
344 flat_value = nest.flatten_up_to(self._nested_structure, value)
349 for sub_value, structure in zip(flat_value, self._flat_nested_structure):
360 flat_value = nest.flatten_up_to(self._nested_structure, value)
365 for sub_value, structure in zip(flat_value, self._flat_nested_structure):
372 def _from_tensor_list(self, flat_value): argument
373 if len(flat_value) != len(self._flat_types):
375 % (len(self._flat_types), len(flat_value)))
[all …]
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | optional_ops.py | 176 def _from_tensor_list(self, flat_value): argument
177 if (len(flat_value) != 1 or flat_value[0].dtype != dtypes.variant or
178 not flat_value[0].shape.is_compatible_with(tensor_shape.scalar())):
181 return self._from_compatible_tensor_list(flat_value)
183 def _from_compatible_tensor_list(self, flat_value): argument
185 return _OptionalImpl(flat_value[0], self._value_structure)
|
| D | dataset_ops.py | 2191 def _from_tensor_list(self, flat_value): argument
2192 if (len(flat_value) != 1 or flat_value[0].dtype != dtypes.variant or
2193 not flat_value[0].shape.is_compatible_with(tensor_shape.scalar())):
2196 return self._from_compatible_tensor_list(flat_value)
2198 def _from_compatible_tensor_list(self, flat_value): argument
2200 return _VariantDataset(flat_value[0], self._element_structure)
|