| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy_test.py | 332 def testMinimizeLossGraph(self, required_gpus): argument
338 def testVariableInitialization(self, required_gpus): argument
347 def testMakeInputFnIterator(self, required_gpus, use_dataset): argument
348 def _worker_fn(task_type, task_id, required_gpus): argument
415 def testMinimizeLossGraph(self, required_gpus): argument
421 def testVariableInitialization(self, required_gpus): argument
436 def testMinimizeLoss(self, required_gpus): argument
447 def testMakeInputFnIterator(self, required_gpus, use_dataset): argument
475 def testAllReduceSum(self, required_gpus): argument
482 def testAllReduceSumGradients(self, required_gpus): argument
[all …]
|
| D | cross_device_ops_test.py | 322 def testAllReduceDense(self, num_processes, required_gpus, implementation, argument
364 def testAllReduceSparse(self, num_processes, required_gpus, implementation, argument
444 def testBatchAllReduceDense(self, num_processes, required_gpus, argument
488 def testBatchAllReduceSparse(self, num_processes, required_gpus, argument
574 def testCollectiveAllReduce(self, num_processes, required_gpus, argument
638 def testAllGatherSameShape(self, num_processes, required_gpus, implementation, argument
679 def testCollectiveV2ControlFlow(self, num_processes, required_gpus, argument
714 required_gpus, argument
775 def testInputsAreFunctionArgs(self, num_processes, required_gpus, argument
817 def testTimeoutReduceDense(self, num_processes, implementation, required_gpus, argument
[all …]
|
| D | parameter_server_strategy_test.py | 623 def testLocalSimpleIncrement(self, required_gpus): argument
628 def testMinimizeLossGraphDistributed(self, required_gpus): argument
634 def testMinimizeLossGraphLocal(self, required_gpus): argument
641 def testMakeInputFnIteratorDistributed(self, required_gpus, use_dataset): argument
671 def testMakeInputFnIteratorLocal(self, required_gpus, use_dataset): argument
833 def testSimpleBetweenGraph(self, required_gpus): argument
|
| D | strategy_combinations.py | 128 def _get_multi_worker_mirrored_creator(required_gpus): argument
|
| D | combinations.py | 256 required_gpus=None, argument
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 729 def testReductionDistributed(self, required_gpus, use_strategy_object, argument
745 def testReduceIndexedSlicesDistributed(self, required_gpus, variable_length): argument
763 def testReductionLocal(self, required_gpus, communication, argument
782 def testReduceIndexedSlicesLocal(self, required_gpus, batch_reduce, argument
|
| /external/tensorflow/tensorflow/tools/docs/ |
| D | tf_doctest.py | 132 def setup_gpu(required_gpus): argument
|