| /external/tensorflow/tensorflow/lite/nnapi/ |
| D | nnapi_handler_test.cc | 41 const auto device_count_stub = [](uint32_t* device_count) -> int { in TEST_F()
50 uint32_t device_count = 0; in TEST_F() local
58 auto device_count_override = [](uint32_t* device_count) -> int { in TEST_F()
79 [](uint32_t* device_count) -> int { in TEST_F()
84 uint32_t device_count = 0; in TEST_F() local
|
| /external/mesa3d/src/vulkan/device-select-layer/ |
| D | device_select.h | 39 …vice_select_find_xcb_pci_default(struct device_pci_info *devices, uint32_t device_count) { return … in device_select_find_xcb_pci_default()
45 …_select_find_wayland_pci_default(struct device_pci_info *devices, uint32_t device_count) { return … in device_select_find_wayland_pci_default()
|
| D | device_select_layer.c | 267 uint32_t device_count, in device_select_find_explicit_default()
285 uint32_t device_count, in device_select_find_dri_prime_tag_default()
305 uint32_t device_count) in device_select_find_boot_vga_default()
329 uint32_t device_count) in device_select_find_non_cpu()
344 uint32_t device_count, in find_non_cpu_skip()
|
| D | device_select_x11.c | 63 int device_select_find_xcb_pci_default(struct device_pci_info *devices, uint32_t device_count) in device_select_find_xcb_pci_default()
|
| D | device_select_wayland.c | 99 int device_select_find_wayland_pci_default(struct device_pci_info *devices, uint32_t device_count) in device_select_find_wayland_pci_default()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tpu_device.cc | 316 int device_count = platform->VisibleDeviceCount(); in ListPhysicalDevices() local
362 int device_count = platform->VisibleDeviceCount(); in CreateDevices() local
407 int device_count = 0; in ListPhysicalDevices() local
422 int device_count = 0; in CreateDevices() local
|
| D | xla_gpu_device.cc | 81 int device_count = platform.ValueOrDie()->VisibleDeviceCount(); in ListPhysicalDevices() local
|
| D | xla_kernel_creator_test.cc | 63 auto* device_count = options.config.mutable_device_count(); in Init() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 99 int device_count() const { return stream_executors_.size(); } in device_count() function
|
| D | platform_util.cc | 168 int device_count = platform->VisibleDeviceCount(); in GetStreamExecutors() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_resolver_local_test.cc | 35 auto* device_count = options.config.mutable_device_count(); in DeviceResolverLocalTest() local
|
| D | collective_executor_mgr_test.cc | 39 auto* device_count = options.config.mutable_device_count(); in CollectiveExecutorMgrTest() local
|
| D | function_threadpool_test.cc | 54 auto* device_count = options.config.mutable_device_count(); in Init() local
|
| D | partitioning_utils_test.cc | 44 auto* device_count = options.config.mutable_device_count(); in SetUp() local
|
| D | collective_rma_local_test.cc | 45 auto* device_count = options.config.mutable_device_count(); in CollectiveRemoteAccessLocalTest() local
|
| D | hierarchical_tree_broadcaster.cc | 131 int device_count = 0; in InitializeCollectiveParams() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.h | 103 int device_count() const { return pjrt_client_->device_count(); } in device_count() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | platform.cc | 110 int device_count = VisibleDeviceCount(); in GetPeerAccessMap() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 56 void TestWithDeviceCount(const int device_count) { in TestWithDeviceCount()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr_test.cc | 46 auto* device_count = options.config.mutable_device_count(); in RpcCollectiveExecutorMgrTest() local
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | profile_analyzer_cli.py | 477 self, device_name, device_index, device_count, argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 109 int device_count() const { return devices_.size(); } in device_count() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 336 int LocalClient::device_count() const { in device_count() function in xla::LocalClient
|
| D | client.cc | 366 int64 device_count) { in GetDeviceHandles()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.h | 135 int device_count() const override { return devices_.size(); } in device_count() function
|