Searched refs:test_prefetch_to_device_dataset (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | one_device_strategy_test.py | 122 def test_prefetch_to_device_dataset(self, distribution): member in OneDeviceStrategyTest |
| D | collective_all_reduce_strategy_test.py | 286 def test_prefetch_to_device_dataset(self, prefetch_to_device): member in DistributedCollectiveAllReduceStrategyTest |
| D | parameter_server_strategy_test.py | 775 def test_prefetch_to_device_dataset(self, prefetch_to_device): member in ParameterServerStrategyTest |
| D | mirrored_strategy_test.py | 237 def test_prefetch_to_device_dataset(self, distribution): member in MirroredTwoDeviceDistributionTest |