| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy_test.py | 353 def testMinimizeLossGraph(self, required_gpus): argument
359 def testVariableInitialization(self, required_gpus): argument
368 def testMakeInputFnIterator(self, required_gpus, use_dataset): argument
369 def _worker_fn(task_type, task_id, required_gpus): argument
436 def testMinimizeLossGraph(self, required_gpus): argument
442 def testVariableInitialization(self, required_gpus): argument
470 def testMinimizeLoss(self, required_gpus, use_devices_arg): argument
482 def testNumReplicasInSync(self, required_gpus, use_devices_arg): argument
502 def testCallAndMergeExceptions(self, required_gpus, use_devices_arg): argument
513 def testMakeInputFnIterator(self, required_gpus, use_dataset, argument
[all …]
|
| D | cross_device_ops_test.py | 325 def testReduceDense(self, num_processes, required_gpus, implementation, argument
372 def testReduceSparse(self, num_processes, required_gpus, implementation, argument
457 def testBatchReduceDense(self, num_processes, required_gpus, implementation, argument
505 def testBatchReduceSparse(self, num_processes, required_gpus, implementation, argument
634 def testAllReduceDense(self, num_processes, required_gpus, implementation, argument
702 def testAllReduceSparse(self, num_processes, required_gpus, implementation, argument
781 def testAllReduceMixedDenseAndSparse(self, num_processes, required_gpus, argument
834 def testAllGatherSameShape(self, num_processes, required_gpus, implementation, argument
888 def testCollectiveV2ControlFlow(self, num_processes, required_gpus, argument
924 required_gpus, argument
[all …]
|
| D | parameter_server_strategy_test.py | 619 def testLocalSimpleIncrement(self, required_gpus): argument
624 def testMinimizeLossGraphDistributed(self, required_gpus): argument
630 def testMinimizeLossGraphLocal(self, required_gpus): argument
637 def testMakeInputFnIteratorDistributed(self, required_gpus, use_dataset): argument
667 def testMakeInputFnIteratorLocal(self, required_gpus, use_dataset): argument
831 def testSimpleBetweenGraph(self, required_gpus): argument
|
| D | strategy_combinations.py | 148 def _get_multi_worker_mirrored_creator(required_gpus, use_merge_call=True): argument
216 required_gpus=0, argument
485 required_gpus=0, argument
|
| D | combinations.py | 282 required_gpus=None, argument
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 733 def testReductionDistributed(self, required_gpus, use_strategy_object, argument
749 def testReduceIndexedSlicesDistributed(self, required_gpus, variable_length): argument
767 def testReductionLocal(self, required_gpus, communication, argument
786 def testReduceIndexedSlicesLocal(self, required_gpus, batch_reduce, argument
|
| /external/tensorflow/tensorflow/tools/docs/ |
| D | tf_doctest.py | 118 def setup_gpu(required_gpus): argument
|