Home
last modified time | relevance | path

Searched refs:device_count (Results 1 – 25 of 97) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/nnapi/
Dnnapi_handler_test.cc41 const auto device_count_stub = [](uint32_t* device_count) -> int { in TEST_F() argument
42 *device_count = 999; in TEST_F()
50 uint32_t device_count = 0; in TEST_F() local
51 nnapi->ANeuralNetworks_getDeviceCount(&device_count); in TEST_F()
52 EXPECT_THAT(device_count, Eq(999)); in TEST_F()
58 auto device_count_override = [](uint32_t* device_count) -> int { in TEST_F() argument
59 *device_count = 777; in TEST_F()
79 [](uint32_t* device_count) -> int { in TEST_F() argument
84 uint32_t device_count = 0; in TEST_F() local
85 EXPECT_THAT(nnapi->ANeuralNetworks_getDeviceCount(&device_count), Eq(1)); in TEST_F()
/external/mesa3d/src/vulkan/device-select-layer/
Ddevice_select.h37 int device_select_find_xcb_pci_default(struct device_pci_info *devices, uint32_t device_count);
39 …_select_find_xcb_pci_default(struct device_pci_info *devices, uint32_t device_count) { return -1; } in device_select_find_xcb_pci_default() argument
43 int device_select_find_wayland_pci_default(struct device_pci_info *devices, uint32_t device_count);
45 …ect_find_wayland_pci_default(struct device_pci_info *devices, uint32_t device_count) { return -1; } in device_select_find_wayland_pci_default() argument
Ddevice_select_layer.c267 uint32_t device_count, in device_select_find_explicit_default() argument
276 for (unsigned i = 0; i < device_count; ++i) { in device_select_find_explicit_default()
285 uint32_t device_count, in device_select_find_dri_prime_tag_default() argument
289 for (unsigned i = 0; i < device_count; ++i) { in device_select_find_dri_prime_tag_default()
305 uint32_t device_count) in device_select_find_boot_vga_default() argument
309 for (unsigned i = 0; i < device_count; ++i) { in device_select_find_boot_vga_default()
329 uint32_t device_count) in device_select_find_non_cpu() argument
334 for (unsigned i = 0; i < device_count; ++i) { in device_select_find_non_cpu()
344 uint32_t device_count, in find_non_cpu_skip() argument
347 for (unsigned i = 0; i < device_count; ++i) { in find_non_cpu_skip()
Ddevice_select_x11.c63 int device_select_find_xcb_pci_default(struct device_pci_info *devices, uint32_t device_count) in device_select_find_xcb_pci_default() argument
99 for (unsigned i = 0; i < device_count; i++) { in device_select_find_xcb_pci_default()
Ddevice_select_wayland.c99 int device_select_find_wayland_pci_default(struct device_pci_info *devices, uint32_t device_count) in device_select_find_wayland_pci_default() argument
125 for (unsigned i = 0; i < device_count; i++) { in device_select_find_wayland_pci_default()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc56 void TestWithDeviceCount(const int device_count) { in TestWithDeviceCount() argument
76 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
87 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
93 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
100 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
105 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
/external/tensorflow/tensorflow/compiler/xla/service/
Dplatform_util.cc168 int device_count = platform->VisibleDeviceCount(); in GetStreamExecutors() local
169 if (device_count <= 0) { in GetStreamExecutors()
179 device_count = in GetStreamExecutors()
182 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr); in GetStreamExecutors()
186 tensorflow::Env::Default(), "device_initialization", device_count); in GetStreamExecutors()
187 for (int i = 0; i < device_count; ++i) { in GetStreamExecutors()
Dbackend.h99 int device_count() const { return stream_executors_.size(); } in device_count() function
136 return (device_ordinal >= 0 && device_ordinal < device_count() && in device_ordinal_supported()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tpu_device.cc316 int device_count = platform->VisibleDeviceCount(); in ListPhysicalDevices() local
318 for (int i = 0; i < device_count; ++i) { in ListPhysicalDevices()
362 int device_count = platform->VisibleDeviceCount(); in CreateDevices() local
363 VLOG(1) << "Creating " << device_count << " TPU devices"; in CreateDevices()
364 for (int i = 0; i < device_count; ++i) { in CreateDevices()
407 int device_count = 0; in ListPhysicalDevices() local
408 TF_RETURN_IF_ERROR(tpu::TpuPlatform::TpusPerHost(&device_count)); in ListPhysicalDevices()
409 if (device_count == 0) { in ListPhysicalDevices()
422 int device_count = 0; in CreateDevices() local
423 TF_RETURN_IF_ERROR(tpu::TpuPlatform::TpusPerHost(&device_count)); in CreateDevices()
[all …]
Dxla_gpu_device.cc81 int device_count = platform.ValueOrDie()->VisibleDeviceCount(); in ListPhysicalDevices() local
82 if (device_count <= 0) { in ListPhysicalDevices()
86 for (int i = 0; i < device_count; ++i) { in ListPhysicalDevices()
130 auto iter = session_options.config.device_count().find("GPU"); in CreateDevices()
131 if (iter != session_options.config.device_count().end() && in CreateDevices()
/external/skqp/src/compute/common/cl/
Dfind_cl.c87 cl_uint device_count; in clFindIdsByName() local
94 &device_count); in clFindIdsByName()
96 cl_device_id * const device_ids = ALLOCA_MACRO(sizeof(*device_ids) * device_count); in clFindIdsByName()
100 device_count, in clFindIdsByName()
107 for (cl_uint jj=0; jj<device_count; jj++) in clFindIdsByName()
/external/autotest/client/common_lib/cros/manual/
Dcfm_helper.py113 for pid_vid, device_count in device_list.iteritems():
114 if device_count > 0:
115 peripheral_map[pid_vid] = device_count
117 for pid_vid, device_count in peripheral_map.iteritems():
120 device_count)
178 for pid_vid, device_count in peripheral_map.iteritems():
179 if device_count > 1:
182 device_count)
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_testlib.cc53 auto iter = options.config.device_count().find("CPU"); in MakeTestCluster()
54 if (iter != options.config.device_count().end()) { in MakeTestCluster()
57 iter = options.config.device_count().find("GPU"); in MakeTestCluster()
58 if (iter != options.config.device_count().end()) { in MakeTestCluster()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmulti_device_iterator_test.py55 config = config_pb2.ConfigProto(device_count={"CPU": 3})
72 config = config_pb2.ConfigProto(device_count={"CPU": 3})
91 config = config_pb2.ConfigProto(device_count={"CPU": 2})
110 config = config_pb2.ConfigProto(device_count={"CPU": 3})
134 config = config_pb2.ConfigProto(device_count={"CPU": 3})
162 config = config_pb2.ConfigProto(device_count={"CPU": 3})
191 config = config_pb2.ConfigProto(device_count={"CPU": 3})
220 config = config_pb2.ConfigProto(device_count={"CPU": 3})
254 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
275 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
[all …]
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_test.py65 config = config_pb2.ConfigProto(device_count={device_type: group_size})
98 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
188 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
249 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
307 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
336 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
378 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
402 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
425 device_count={'CPU': group_size})) as sess:
528 cfg = config_pb2.ConfigProto(device_count={'CPU': group_size},
/external/tensorflow/tensorflow/stream_executor/
Dplatform.cc110 int device_count = VisibleDeviceCount(); in GetPeerAccessMap() local
111 for (int i = 0; i < device_count; ++i) { in GetPeerAccessMap()
112 for (int j = 0; j < device_count; ++j) { in GetPeerAccessMap()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dprefetch_to_device_test.py56 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
81 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
105 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
132 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
181 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
Dcopy_to_device_test.py60 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
85 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
109 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
134 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
159 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
184 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
214 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
247 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
440 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
469 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_resolver_local_test.cc35 auto* device_count = options.config.mutable_device_count(); in DeviceResolverLocalTest() local
36 device_count->insert({"CPU", NUM_DEVS}); in DeviceResolverLocalTest()
Dthreadpool_device_factory.cc42 auto iter = options.config.device_count().find("CPU"); in CreateDevices()
43 if (iter != options.config.device_count().end()) { in CreateDevices()
Dcollective_executor_mgr_test.cc39 auto* device_count = options.config.mutable_device_count(); in CollectiveExecutorMgrTest() local
41 device_count->insert({"CPU", NUM_DEVS}); in CollectiveExecutorMgrTest()
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc58 auto iter = options.config.device_count().find("CPU"); in MakeGRPCCluster()
59 if (iter != options.config.device_count().end()) { in MakeGRPCCluster()
62 iter = options.config.device_count().find("GPU"); in MakeGRPCCluster()
63 if (iter != options.config.device_count().end()) { in MakeGRPCCluster()
/external/tensorflow/tensorflow/tools/test/
Dgpu_info_lib.py144 device_count = ct.c_int()
145 libcudart.cudaGetDeviceCount(ct.byref(device_count))
147 for i in range(device_count.value):
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli.py417 device_count = len(self._run_metadata.step_stats.dev_stats)
418 for index in range(device_count):
433 device_stats.device, index, device_count,
477 self, device_name, device_index, device_count, argument
536 device_index + 1, device_count, device_name)
622 device_count = len(self._run_metadata.step_stats.dev_stats)
623 for index in range(device_count):
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtpu_client.cc144 for (int i = 0; i < client->device_count(); ++i) { in GetTpuDevices()
207 local_device_states.reserve(client->device_count()); in GetTpuClient()
208 for (int i = 0; i < client->device_count(); ++i) { in GetTpuClient()

1234