Home
last modified time | relevance | path

Searched refs:num_gpus (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/all_reduce/python/
Dall_reduce_test.py82 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 …]
Dall_reduce.py223 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/
Dbenchmarks_test.py129 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 …]
Dcore_test.py118 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 …]
Dops_test.py48 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():
Dcontext.py338 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/
Drnn_ptb_test.py32 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/
Dgrpc_testlib_server.cc39 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()
Dgrpc_testlib.cc42 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/
Dtraining_utils.py122 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/
Ddist_mnist_test.sh134 --num_gpus=${N_GPUS} \
159 --num_gpus=${N_GPUS} \
189 --num_gpus=${N_GPUS} \
/external/tensorflow/tensorflow/python/kernel_tests/
Dlist_ops_test.py52 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/
Drpcbench_test.cc57 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/
Dsi_gb_reg.h56 unsigned int num_gpus : 3; member
74 unsigned int num_gpus : 3; member
/external/tensorflow/tensorflow/core/grappler/
Ddevices.cc35 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/
Dsingle_machine.cc39 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/
Dmnist_test.py27 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/
Dmnist_replica.py119 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/
Dmnist_test.py40 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/
Dlinear_regression_test.py34 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/
Dperformance_models.md345 * **`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/
Drnn_colorbot.py240 if FLAGS.no_gpu or tfe.num_gpus() <= 0:
241 print(tfe.num_gpus())
/external/tensorflow/tensorflow/core/debug/
Dgrpc_session_debug_test.cc45 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/
Dlocal_test.sh163 --num_gpus 0 ${SYNC_REPLICAS_FLAG}
/external/tensorflow/tensorflow/contrib/eager/python/examples/spinn/
Dspinn_test.py61 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"

12