Searched refs:_experimental_distribute_dataset (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | distribute_lib.py | 1111 …return self._extended._experimental_distribute_dataset(dataset, options) # pylint: disable=protec…
2287 def _experimental_distribute_dataset(self, dataset, options): member in StrategyExtendedV2
3654 def _experimental_distribute_dataset(self, dataset, options): member in _DefaultDistributionExtended
|
| H A D | one_device_strategy.py | 310 def _experimental_distribute_dataset(self, dataset, options): member in OneDeviceExtended
|
| H A D | parameter_server_strategy_v2.py | 870 def _experimental_distribute_dataset(self, dataset, options): member in ParameterServerStrategyV2Extended
|
| H A D | parameter_server_strategy.py | 339 def _experimental_distribute_dataset(self, dataset, options): member in ParameterServerStrategyExtended
|
| H A D | mirrored_strategy.py | 564 def _experimental_distribute_dataset(self, dataset, options): member in MirroredExtended
|
| H A D | collective_all_reduce_strategy.py | 649 def _experimental_distribute_dataset(self, dataset, options): member in CollectiveAllReduceExtended
|
| H A D | tpu_strategy.py | 990 def _experimental_distribute_dataset(self, dataset, options): member in TPUExtended
|