Home
last modified time | relevance | path

Searched refs:_get_num_gpus (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dremote_mirrored_strategy_eager_test.py52 def _get_num_gpus(self): member in RemoteSingleWorkerMirroredStrategyEager
Dstrategy_test_lib.py728 def _get_num_gpus(self): member in RemoteSingleWorkerMirroredStrategyBase
732 self.assertEqual(self._get_num_gpus(), distribution.num_replicas_in_sync)
754 num_gpus = self._get_num_gpus()
778 num_gpus = self._get_num_gpus()
Dmirrored_strategy_test.py1261 def _get_num_gpus(self): member in RemoteSingleWorkerMirroredStrategyGraph