Home
last modified time | relevance | path

Searched refs:get_flat_tensor_shapes (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Diterator_ops.py138 self._flat_tensor_shapes = structure.get_flat_tensor_shapes(
237 output_shapes=structure.get_flat_tensor_shapes(
309 output_shapes=structure.get_flat_tensor_shapes(output_structure))
460 output_shapes=structure.get_flat_tensor_shapes(
695 self._flat_output_shapes = structure.get_flat_tensor_shapes(
721 self._flat_output_shapes = structure.get_flat_tensor_shapes(
850 output_shapes=structure.get_flat_tensor_shapes(
H A Dmulti_device_iterator_ops.py76 output_shapes=structure.get_flat_tensor_shapes(
83 output_shapes=structure.get_flat_tensor_shapes(self._element_spec))
H A Doptional_ops.py185 output_shapes=structure.get_flat_tensor_shapes(self._element_spec))
H A Dsave_op.py88 output_shapes=structure.get_flat_tensor_shapes(dataset.element_spec),
H A Ddataset_ops.py634 return structure.get_flat_tensor_shapes(self.element_spec)
2686 output_shapes=structure.get_flat_tensor_shapes(state_structure),
4702 output_shapes=structure.get_flat_tensor_shapes(self._structure),
4734 output_shapes=structure.get_flat_tensor_shapes(self._structure),
5383 output_shapes=structure.get_flat_tensor_shapes(self._structure),
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Drandom_access.py73 output_shapes=structure.get_flat_tensor_shapes(dataset.element_spec)))
H A Dcompression_ops.py48 flat_shapes = structure.get_flat_tensor_shapes(output_spec)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dstructure.py292 def get_flat_tensor_shapes(element_spec): function
H A Dstructure_test.py513 flat_shapes = structure.get_flat_tensor_shapes(s)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Doptional_test.py338 structure.get_flat_tensor_shapes(opt_structure))
H A Ddataset_test.py329 structure.get_flat_tensor_shapes(dataset_structure))