| /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 | 324 int device_count = platform->VisibleDeviceCount(); in ListPhysicalDevices() local
370 int device_count = platform->VisibleDeviceCount(); in CreateDevices() local
420 int device_count = 0; in ListPhysicalDevices() local
435 int device_count = 0; in CreateDevices() local
|
| D | xla_gpu_device.cc | 61 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/core/tfrt/fallback/ |
| D | op_kernel_runner_test.cc | 44 auto* device_count = options.config.mutable_device_count(); in TEST() local
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_factory.cc | 125 int device_count = platform->VisibleDeviceCount(); in ListPhysicalDevices() local
143 int device_count = platform->VisibleDeviceCount(); in GetDeviceDetails() local
|
| /external/tensorflow/tensorflow/core/util/autotune_maps/ |
| D | autotune_maps_utils.cc | 62 int device_count = GpuDriver::GetDeviceCount(); in GetDeviceIdToIdentifierMap() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 100 int device_count() const { return stream_executors_.size(); } in device_count() function
|
| D | platform_util.cc | 160 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 | 45 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
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | platform.cc | 109 int device_count = VisibleDeviceCount(); in GetPeerAccessMap() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.h | 112 int device_count() const { return pjrt_client_->device_count(); } in device_count() function
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 73 const auto& device_count = config.device_count(); in MaybeCreateNcclCommunicator() 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/tfrt/utils/ |
| D | graph_partition_test.cc | 50 auto* device_count = options.config.mutable_device_count(); in SetUp() local
|
| /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/virglrenderer/src/venus/ |
| D | vkr_allocator.c | 60 uint32_t device_count; member
|
| /external/tensorflow/tensorflow/core/data/ |
| D | serialization_utils_test.cc | 54 auto* device_count = options.config.mutable_device_count(); in Create() local
|
| /external/crosvm/devices/src/ |
| D | bus_stats.rs | 198 let device_count = stats[0].lock().device_identifiers.lock().len(); in merged() localVariable
|