Searched refs:test_prefetch_to_device_dataset (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | one_device_strategy_test.py | 119 def test_prefetch_to_device_dataset(self, distribution): member in OneDeviceStrategyTest |
| H A D | collective_all_reduce_strategy_test.py | 311 def test_prefetch_to_device_dataset(self, prefetch_to_device): member in DistributedCollectiveAllReduceStrategyTest |
| H A D | parameter_server_strategy_test.py | 771 def test_prefetch_to_device_dataset(self, prefetch_to_device): member in ParameterServerStrategyTest |
| H A D | mirrored_strategy_test.py | 261 def test_prefetch_to_device_dataset(self, distribution): member in MirroredTwoDeviceDistributionTest |