| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_spec_test.py | 176 self.assertEqual(batched._unbatch(), unbatched)
242 unbatched_spec = type_spec.type_spec_from_value(batched)._unbatch()
261 batched_spec._unbatch()._from_tensor_list(tensor_list)
265 self.assertTrue(batched_spec._unbatch().is_compatible_with(st))
|
| D | structured_tensor.py | 1094 def _unbatch(self): member in StructuredTensorSpec
1098 dict((k, v._unbatch()) for (k, v) in self._field_specs.items()))
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | distribute.py | 142 lambda ts: ts._unbatch()._batch(new_batch_dim),
280 return type_spec._unbatch()._batch(batch_size)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | map_fn.py | 393 _most_general_compatible_type(s)._unbatch() # pylint: disable=protected-access
578 spec = spec._unbatch()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_spec.py | 216 def _unbatch(self): member in TensorSpec
|
| D | sparse_tensor.py | 404 def _unbatch(self): member in SparseTensorSpec
|
| D | type_spec.py | 518 def _unbatch(self): member in BatchableTypeSpec
|
| D | sparse_tensor_test.py | 355 self.assertEqual(spec._unbatch(), expected)
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 526 def _unbatch(self): member in NoneTensorSpec
|
| D | structure_test.py | 688 lambda component_spec: component_spec._unbatch(), element_structure)
725 lambda component_spec: component_spec._unbatch(), s)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops.py | 418 return spec._unbatch()._from_compatible_tensor_list(gathered_tensors)
|
| D | control_flow_ops_test.py | 2134 def _unbatch(self): member in ParticleSpec
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 3214 …lambda component_spec: component_spec._unbatch(), batched_spec) # pylint: disable=protected-access
3377 def _unbatch(self): member in DatasetSpec
4821 lambda component_spec: component_spec._unbatch(), # pylint: disable=protected-access
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_test.py | 1878 first_row = rt_spec._unbatch()._from_tensor_list(
1915 self.assertEqual(spec._unbatch(), expected)
|
| D | ragged_tensor.py | 2461 def _unbatch(self): member in RaggedTensorSpec
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 2206 return spec._unbatch()._batch(None)
|