/external/tensorflow/tensorflow/contrib/all_reduce/python/ |
D | all_reduce_test.py | 82 def _buildInput(self, num_workers, num_gpus): argument 89 for d in range(0, num_gpus): 90 dn = "/replica:0/task:%d/device:GPU:%d" % (w, d % num_gpus) 136 def _buildRing(self, num_workers, num_gpus, subdiv): argument 137 gpu_perm = range(0, num_gpus) 141 def _testAllReduce(self, num_workers, num_gpus, shape, build_f): argument 143 num_devices = num_workers * num_gpus 157 def _testRingAllReduce(self, num_workers, num_gpus, shape, subdiv): argument 159 build_f = self._buildRing(num_workers, num_gpus, subdiv) 160 self._testAllReduce(num_workers, num_gpus, shape, build_f) [all …]
|
D | all_reduce.py | 223 num_gpus = len(gpu_perm) 224 devices = num_workers * num_gpus 227 if num_subchunks > num_gpus: 229 "num_subchunks %d must be <= num_gpus %d" % (num_subchunks, num_gpus)) 230 rotation_interval = max(1, int(num_gpus / num_subchunks)) 236 default_order = [(w * num_gpus) + i for i in gpu_perm]
|
/external/tensorflow/tensorflow/python/eager/ |
D | benchmarks_test.py | 129 if not context.num_gpus(): 134 if not context.num_gpus(): 142 if not context.num_gpus(): 148 if not context.num_gpus(): 175 if not context.num_gpus(): 187 if not context.num_gpus(): 319 if not context.num_gpus(): 327 if not context.num_gpus(): 335 if not context.num_gpus(): 343 if not context.num_gpus(): [all …]
|
D | core_test.py | 118 if not context.context().num_gpus(): 141 ctx.device_name, ctx.num_gpus() 153 if not context.context().num_gpus(): 157 self.assertEquals(0, ctx.num_gpus()) 160 if not context.context().num_gpus(): 173 if not context.context().num_gpus(): 184 x.gpu(context.context().num_gpus() + 1) 187 if not context.context().num_gpus(): 196 if not context.context().num_gpus(): 237 if not context.context().num_gpus(): [all …]
|
D | ops_test.py | 48 if not context.context().num_gpus(): 97 if not context.context().num_gpus(): 238 if not context.context().num_gpus(): 256 if not context.context().num_gpus(): 264 if not context.context().num_gpus(): 318 if not context.context().num_gpus():
|
D | context.py | 338 def num_gpus(self): member in Context 571 def num_gpus(): function 577 return context().num_gpus()
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/ |
D | rnn_ptb_test.py | 32 return "/device:GPU:0" if tfe.num_gpus() else "/device:CPU:0" 38 model = rnn_ptb.test_model(tfe.num_gpus() > 0) 48 model = rnn_ptb.test_model(tfe.num_gpus() > 0) 54 if tfe.num_gpus(): 98 if not tfe.num_gpus(): 103 if not tfe.num_gpus(): 142 if not tfe.num_gpus(): 147 if not tfe.num_gpus():
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
D | grpc_testlib_server.cc | 39 int num_cpus, int num_gpus, int task_index, in FillServerDef() argument 71 (*config->mutable_device_count())["GPU"] = num_gpus; in FillServerDef() 83 int num_gpus = 0; in main() local 90 tensorflow::Flag("num_gpus", &num_gpus, "number of GPUs"), in main() 101 num_gpus, task_index, &def); in main()
|
D | grpc_testlib.cc | 42 int num_gpus = 0; in MakeTestCluster() local 49 num_gpus = iter->second; in MakeTestCluster() 59 strings::StrCat("--num_gpus=", num_gpus)}); in MakeTestCluster()
|
/external/tensorflow/tensorflow/python/keras/_impl/keras/utils/ |
D | training_utils.py | 122 num_gpus = len(gpus) 129 num_gpus = gpus 130 target_gpu_ids = range(num_gpus) 160 if i == num_gpus - 1: 187 'parts': num_gpus
|
/external/tensorflow/tensorflow/tools/dist_test/scripts/ |
D | dist_mnist_test.sh | 134 --num_gpus=${N_GPUS} \ 159 --num_gpus=${N_GPUS} \ 189 --num_gpus=${N_GPUS} \
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | list_ops_test.py | 52 if not context.num_gpus(): 66 if not context.num_gpus(): 81 if not context.num_gpus(): 96 if not context.num_gpus(): 110 if not context.num_gpus():
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | rpcbench_test.cc | 57 int num_gpus = 0; in MakeGRPCCluster() local 64 num_gpus = iter->second; in MakeGRPCCluster() 69 worker_threads->Schedule([worker_idx, n, num_cpus, num_gpus, &port] { in MakeGRPCCluster() 84 (*config->mutable_device_count())["GPU"] = num_gpus; in MakeGRPCCluster()
|
/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/grappler/ |
D | devices.cc | 35 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs() local 36 for (int i = 0; i < num_gpus; i++) { in GetNumAvailableGPUs()
|
/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() argument 42 << " Number of GPUs: " << num_gpus; in SingleMachine() 47 if (num_gpus > 0) { in SingleMachine() 48 (*options_.config.mutable_device_count())["GPU"] = num_gpus; in SingleMachine()
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/mnist/ |
D | mnist_test.py | 27 return "/device:GPU:0" if tfe.num_gpus() else "/device:CPU:0" 31 return "channels_first" if tfe.num_gpus() else "channels_last"
|
/external/tensorflow/tensorflow/tools/dist_test/python/ |
D | mnist_replica.py | 119 if FLAGS.num_gpus > 0: 122 gpu = (FLAGS.task_index % FLAGS.num_gpus) 124 elif FLAGS.num_gpus == 0:
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/gan/ |
D | mnist_test.py | 40 return '/gpu:0' if tfe.num_gpus() else '/cpu:0' 47 dev = 'gpu' if tfe.num_gpus() else 'cpu'
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/linear_regression/ |
D | linear_regression_test.py | 34 return "/device:GPU:0" if tfe.num_gpus() > 0 else "/device:CPU:0" 113 ("gpu" if tfe.num_gpus() > 0 else "cpu"),
|
/external/tensorflow/tensorflow/docs_src/performance/ |
D | performance_models.md | 345 * **`num_gpus`**: Number of GPUs to use. 361 python tf_cnn_benchmarks.py --local_parameter_device=cpu --num_gpus=8 \ 367 python tf_cnn_benchmarks.py --local_parameter_device=gpu --num_gpus=8 \ 372 python tf_cnn_benchmarks.py --local_parameter_device=gpu --num_gpus=8 \ 377 python tf_cnn_benchmarks.py --local_parameter_device=gpu --num_gpus=8 \ 400 python tf_cnn_benchmarks.py --local_parameter_device=gpu --num_gpus=8 \ 405 python tf_cnn_benchmarks.py --local_parameter_device=gpu --num_gpus=8 \ 412 python tf_cnn_benchmarks.py --local_parameter_device=gpu --num_gpus=8 \ 417 python tf_cnn_benchmarks.py --local_parameter_device=gpu --num_gpus=8 \
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_colorbot/ |
D | rnn_colorbot.py | 240 if FLAGS.no_gpu or tfe.num_gpus() <= 0: 241 print(tfe.num_gpus())
|
/external/tensorflow/tensorflow/core/debug/ |
D | grpc_session_debug_test.cc | 45 SessionOptions Devices(int num_cpus, int num_gpus) { in Devices() argument 48 (*result.config.mutable_device_count())["GPU"] = num_gpus; in Devices()
|
/external/tensorflow/tensorflow/tools/dist_test/ |
D | local_test.sh | 163 --num_gpus 0 ${SYNC_REPLICAS_FLAG}
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/spinn/ |
D | spinn_test.py | 61 if tfe.num_gpus(): 123 self._test_device = "gpu:0" if tfe.num_gpus() else "cpu:0" 435 test_device = "gpu:0" if tfe.num_gpus() else "cpu:0"
|