HomeSort by relevance Sort by last modified time
    Searched refs:device_count (Results 1 - 25 of 81) sorted by null

1 2 3 4

  /external/tensorflow/tensorflow/core/common_runtime/sycl/
sycl_device_factory.cc 32 auto iter = options.config.device_count().find("SYCL");
33 if (iter != options.config.device_count().end()) {
  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
grpc_testlib.cc 44 auto iter = options.config.device_count().find("CPU");
45 if (iter != options.config.device_count().end()) {
48 iter = options.config.device_count().find("GPU");
49 if (iter != options.config.device_count().end()) {
  /external/skia/src/compute/common/cl/
find_cl.c 87 cl_uint device_count; local
94 &device_count);
96 cl_device_id * const device_ids = ALLOCA_MACRO(sizeof(*device_ids) * device_count);
100 device_count,
107 for (cl_uint jj=0; jj<device_count; jj++)
  /external/skqp/src/compute/common/cl/
find_cl.c 87 cl_uint device_count; local
94 &device_count);
96 cl_device_id * const device_ids = ALLOCA_MACRO(sizeof(*device_ids) * device_count);
100 device_count,
107 for (cl_uint jj=0; jj<device_count; jj++)
  /external/autotest/client/common_lib/cros/manual/
cfm_helper.py 110 for pid_vid, device_count in speaker_list.iteritems():
111 if device_count > 0:
112 peripheral_map[pid_vid] = device_count
114 for pid_vid, device_count in camera_list.iteritems():
115 if device_count > 0:
116 peripheral_map[pid_vid] = device_count
118 for pid_vid, device_count in controller_list.iteritems():
119 if device_count > 0:
120 peripheral_map[pid_vid] = device_count
122 for pid_vid, device_count in display_list.iteritems()
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
multiple_devices_on_host_test.cc 55 void TestWithDeviceCount(const int device_count) {
56 // Run `device_count` copies of the XLA program built by BuildComputation.
74 for (int device_ordinal = 0; device_ordinal < device_count;
85 for (int device_ordinal = 0; device_ordinal < device_count;
91 for (int device_ordinal = 0; device_ordinal < device_count;
99 for (int device_ordinal = 0; device_ordinal < device_count;
104 for (int device_ordinal = 0; device_ordinal < device_count;
local_client_allocation_test.cc 83 for (int d = 0; d < local_client_->device_count(); ++d) {
  /external/tensorflow/tensorflow/core/common_runtime/
device_resolver_local_test.cc 36 auto* device_count = options.config.mutable_device_count(); local
37 device_count->insert({"CPU", NUM_DEVS});
threadpool_device_factory.cc 36 auto iter = options.config.device_count().find("CPU");
37 if (iter != options.config.device_count().end()) {
collective_executor_mgr_test.cc 38 auto* device_count = options.config.mutable_device_count(); local
40 device_count->insert({"CPU", NUM_DEVS});
collective_rma_local_test.cc 43 auto* device_count = options.config.mutable_device_count(); local
44 device_count->insert({"CPU", NUM_DEVS});
device_factory.cc 132 auto iter = options.config.device_count().find(type);
133 if (iter != options.config.device_count().end()) {
  /external/tensorflow/tensorflow/stream_executor/
platform.cc 110 int device_count = VisibleDeviceCount(); local
111 for (int i = 0; i < device_count; ++i) {
112 for (int j = 0; j < device_count; ++j) {
  /external/tensorflow/tensorflow/compiler/xla/service/
platform_util.cc 214 int device_count = platform->VisibleDeviceCount(); local
215 if (device_count <= 0) {
225 device_count =
228 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr);
232 tensorflow::Env::Default(), "device_initialization", device_count); local
233 for (int i = 0; i < device_count; ++i) {
backend.h 99 int device_count() const { return stream_executors_.size(); } function in class:xla::Backend
136 return (device_ordinal >= 0 && device_ordinal < device_count() &&
  /external/tensorflow/tensorflow/python/data/kernel_tests/
multi_device_iterator_test.py 48 config = config_pb2.ConfigProto(device_count={"CPU": 3})
59 config = config_pb2.ConfigProto(device_count={"CPU": 3})
78 config = config_pb2.ConfigProto(device_count={"CPU": 2})
97 config = config_pb2.ConfigProto(device_count={"CPU": 3})
121 config = config_pb2.ConfigProto(device_count={"CPU": 3})
149 config = config_pb2.ConfigProto(device_count={"CPU": 3})
178 config = config_pb2.ConfigProto(device_count={"CPU": 3})
207 config = config_pb2.ConfigProto(device_count={"CPU": 3})
239 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
260 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1}
    [all...]
  /external/tensorflow/tensorflow/compiler/jit/
create_xla_launch_op_test.cc 60 auto* device_count = options.config.mutable_device_count(); local
61 device_count->insert({"CPU", 1});
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device_factory.cc 115 auto iter = options.config.device_count().find("CPU");
116 if (iter != options.config.device_count().end()) {
  /external/tensorflow/tensorflow/core/distributed_runtime/
collective_param_resolver_distributed_test.cc 206 const int device_count = num_workers * num_devices; local
211 cp_.resize(device_count);
212 status_.resize(device_count);
223 int device_count = num_workers * num_devices; local
236 [this, idx, device_count](const Status& s) {
241 if (num_done_ == device_count) {
249 int device_count = num_workers * num_devices; local
252 if (num_done_ < device_count) {
rpcbench_test.cc 58 auto iter = options.config.device_count().find("CPU");
59 if (iter != options.config.device_count().end()) {
62 iter = options.config.device_count().find("GPU");
63 if (iter != options.config.device_count().end()) {
  /external/tensorflow/tensorflow/python/ops/
collective_ops_test.py 35 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
56 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
96 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
119 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
  /external/tensorflow/tensorflow/compiler/xla/client/
local_client.h 172 int device_count() const;
  /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
prefetch_to_device_test.py 53 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
103 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
131 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
179 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
copy_to_device_test.py 57 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
82 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
106 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
131 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
156 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
181 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
211 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
244 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
456 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
485 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2}
    [all...]
  /external/tensorflow/tensorflow/tools/test/
gpu_info_lib.py 139 device_count = ct.c_int()
140 libcudart.cudaGetDeviceCount(ct.byref(device_count))
142 for i in range(device_count.value):

Completed in 748 milliseconds

1 2 3 4