| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce_test.py | 90 def _buildInput(self, num_workers, num_gpus): argument
145 def _buildRing(self, num_workers, num_gpus, subdiv): argument
150 def _testAllReduce(self, num_workers, num_gpus, shape, build_f): argument
166 def _testRingAllReduce(self, num_workers, num_gpus, shape, subdiv): argument
190 def _buildShuffle(self, num_workers, num_gpus, num_shards): argument
197 def _testShuffleAllReduce(self, num_workers, num_gpus, shape, num_shards): argument
217 def _buildRecursiveHD(self, num_workers, num_gpus): argument
223 def _testRecursiveHDAllReduce(self, num_workers, num_gpus, shape): argument
|
| D | mirrored_strategy.py | 405 def all_local_devices(num_gpus=None): argument
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy_test.py | 74 num_gpus=None, argument
126 num_gpus=0, argument
150 num_gpus, argument
222 num_gpus, argument
278 num_gpus, argument
311 num_gpus, argument
379 def testMinimizeLossGraph(self, num_gpus, use_core_strategy): argument
392 def testVariableInitialization(self, num_gpus, use_core_strategy): argument
407 def testComplexModel(self, num_gpus, use_core_strategy): argument
425 def DISABLED_testMakeInputFnIterator(self, num_gpus, use_dataset, argument
[all …]
|
| D | parameter_server_strategy_test.py | 82 num_gpus=None, argument
137 num_gpus, argument
150 num_gpus, argument
252 self, task_type, task_id, num_gpus, use_core_strategy=False): argument
304 num_gpus=0): argument
405 num_gpus, argument
479 num_gpus, argument
571 num_gpus, argument
656 def testDeviceAssignmentDistributed(self, num_gpus, use_core_strategy): argument
663 def testDeviceAssignmentDistributedEnablePartitioner(self, num_gpus, argument
[all …]
|
| D | cross_device_ops_test.py | 456 num_gpus=0, argument
531 num_gpus, argument
645 def testReductionDistributed(self, num_gpus, use_strategy_object): argument
661 def testReductionLocal(self, num_gpus, use_strategy_object): argument
|
| D | mirrored_strategy.py | 89 num_gpus=None, argument
|
| D | multi_worker_test_base.py | 282 def _run_client(self, client_fn, task_type, task_id, num_gpus, *args, argument
289 def _run_between_graph_clients(self, client_fn, cluster_spec, num_gpus, *args, argument
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_testlib_server.cc | 39 int num_cpus, int num_gpus, int task_index, in FillServerDef()
83 int num_gpus = 0; in main() local
|
| D | grpc_testlib.cc | 43 int num_gpus = 0; in MakeTestCluster() local
|
| D | grpc_session_test.cc | 37 static SessionOptions Devices(int num_cpus, int num_gpus) { in Devices()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | graph_placer_test.py | 62 def _buildCluster(num_cpus=1, num_gpus=1): argument
|
| D | cluster.i | 136 int num_gpus = tensorflow::grappler::GetNumAvailableGPUs(); in TF_NewCluster() local
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 36 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs() local
|
| /external/mesa3d/src/amd/addrlib/inc/chip/r800/ |
| D | si_gb_reg.h | 64 unsigned int num_gpus : 3; member
82 unsigned int num_gpus : 3; member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpcbench_test.cc | 57 int num_gpus = 0; in MakeGRPCCluster() local
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | grpc_session_debug_test.cc | 45 SessionOptions Devices(int num_cpus, int num_gpus) { in Devices()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 41 SingleMachine::SingleMachine(int timeout_s, int num_cpu_cores, int num_gpus) in SingleMachine()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 665 def num_gpus(self): member in Context
1100 def num_gpus(): function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer_test.cc | 105 int num_gpus = local_devices.size(); in Init() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster_test.cc | 535 int num_gpus) { in PrepColParamsForSubdivPermsTest()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 2171 int num_gpus = 0; in GetNumGPUs() local
|