Searched defs:use_devices_arg (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | collective_all_reduce_strategy_test.py | 115 use_devices_arg=False): argument
246 use_devices_arg=False): argument
470 def testMinimizeLoss(self, required_gpus, use_devices_arg): argument
482 def testNumReplicasInSync(self, required_gpus, use_devices_arg): argument
492 def testMinimizeLossTPU(self, required_tpus, use_devices_arg): argument
502 def testCallAndMergeExceptions(self, required_gpus, use_devices_arg): argument
514 use_devices_arg): argument
546 def testReduceToCpu(self, required_gpus, use_devices_arg): argument
558 def testAllReduceSum(self, required_gpus, use_devices_arg): argument
567 def testAllReduceSumGradients(self, required_gpus, use_devices_arg): argument
[all …]
|