| /aosp_15_r20/external/tensorflow/tensorflow/lite/nnapi/ |
| H A 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
|
| /aosp_15_r20/external/mesa3d/src/vulkan/device-select-layer/ |
| H A D | device_select_layer.c | 270 uint32_t device_count, in device_select_find_explicit_default()
288 uint32_t device_count, in device_select_find_dri_prime_tag_default()
317 uint32_t device_count) in device_select_find_boot_vga_vid_did()
374 uint32_t device_count) in device_select_find_boot_vga_default()
398 uint32_t device_count) in device_select_find_non_cpu()
413 uint32_t device_count, in find_non_cpu_skip()
|
| H A D | device_select.h | 47 …vice_select_find_xcb_pci_default(struct device_pci_info *devices, uint32_t device_count) { return … in device_select_find_xcb_pci_default()
53 …_select_find_wayland_pci_default(struct device_pci_info *devices, uint32_t device_count) { return … in device_select_find_wayland_pci_default()
|
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A 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
|
| H A D | xla_gpu_device.cc | 61 int device_count = platform.ValueOrDie()->VisibleDeviceCount(); in ListPhysicalDevices() local
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | device_interface.py | 83 def device_count(): member in DeviceInterface
195 device_count = staticmethod(torch.cuda.device_count) variable in CudaInterface
266 device_count = staticmethod(torch.xpu.device_count) variable in XpuInterface
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_factory.cc | 125 int device_count = platform->VisibleDeviceCount(); in ListPhysicalDevices() local
143 int device_count = platform->VisibleDeviceCount(); in GetDeviceDetails() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | CUDAPluggableAllocator.cpp | 28 int device_count = 0; variable
163 void CUDAPluggableAllocator::init(int device_count) { in init()
|
| /aosp_15_r20/test/mlts/benchmark/jni/ |
| D | benchmark_jni.cpp | 441 uint32_t device_count = 0; in Java_com_android_nn_benchmark_core_NNTestBase_hasAccelerator() local
455 uint32_t device_count = 0; in Java_com_android_nn_benchmark_core_NNTestBase_getAcceleratorNames() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | op_kernel_runner_test.cc | 44 auto* device_count = options.config.mutable_device_count(); in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/ |
| H A D | xpu.cpp | 6 size_t device_count() { in device_count() function
|
| H A D | cuda.cpp | 12 size_t device_count() { in device_count() function
|
| /aosp_15_r20/external/pytorch/test/distributed/_shard/sharding_plan/ |
| H A D | test_sharding_plan.py | 38 def __init__(self, chunk_dim=0, device_count=0): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| H A D | oneDNNContext.h | 38 int device_count = (int)c10::xpu::device_count(); in GpuEngineManager() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/xpu/detail/ |
| H A D | XPUHooks.cpp | 14 const auto device_count = c10::xpu::device_count_ensure_non_zero(); in initXPU() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/ |
| H A D | autotune_maps_utils.cc | 62 int device_count = GpuDriver::GetDeviceCount(); in GetDeviceIdToIdentifierMap() local
|
| /aosp_15_r20/external/pytorch/torch/mps/ |
| H A D | __init__.py | 26 def device_count() -> int: function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | backend.h | 100 int device_count() const { return stream_executors_.size(); } in device_count() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | device_resolver_local_test.cc | 35 auto* device_count = options.config.mutable_device_count(); in DeviceResolverLocalTest() local
|
| H A D | collective_executor_mgr_test.cc | 45 auto* device_count = options.config.mutable_device_count(); in CollectiveExecutorMgrTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | platform.cc | 109 int device_count = VisibleDeviceCount(); in GetPeerAccessMap() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | collective_communicator.cc | 73 const auto& device_count = config.device_count(); in MaybeCreateNcclCommunicator() local
|
| /aosp_15_r20/external/mesa3d/src/tool/pps/ |
| H A D | pps_device.cc | 22 uint32_t DrmDevice::device_count() in device_count() function in pps::DrmDevice
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | py_client.h | 112 int device_count() const { return pjrt_client_->device_count(); } in device_count() function
|