| /external/tensorflow/tensorflow/contrib/all_reduce/python/ |
| D | all_reduce_test.py | 82 def _buildInput(self, num_workers, num_gpus): argument
136 def _buildRing(self, num_workers, num_gpus, subdiv): argument
141 def _testAllReduce(self, num_workers, num_gpus, shape, build_f): argument
157 def _testRingAllReduce(self, num_workers, num_gpus, shape, subdiv): argument
180 def _buildShuffle(self, num_workers, num_gpus, num_shards): argument
187 def _testShuffleAllReduce(self, num_workers, num_gpus, shape, num_shards): argument
206 def _buildRecursiveHD(self, num_workers, num_gpus): argument
212 def _testRecursiveHDAllReduce(self, num_workers, num_gpus, shape): 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 | 42 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/core/grappler/ |
| D | devices.cc | 35 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 338 def num_gpus(self): member in Context
571 def num_gpus(): function
|
| /external/mesa3d/src/amd/addrlib/inc/chip/r800/ |
| D | si_gb_reg.h | 56 unsigned int num_gpus : 3; member
74 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/grappler/clusters/ |
| D | single_machine.cc | 39 SingleMachine::SingleMachine(int timeout_s, int num_cpu_cores, int num_gpus) in SingleMachine()
|
| /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/python/grappler/ |
| D | cluster.i | 136 int num_gpus = tensorflow::grappler::GetNumAvailableGPUs(); in TF_NewCluster() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 2085 int num_gpus = 0; in GetNumGPUs() local
|