| /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 | 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()
|
| D | device_select_layer.c | 273 uint32_t device_count, in device_select_find_explicit_default()
291 uint32_t device_count, in device_select_find_dri_prime_tag_default()
320 uint32_t device_count) in device_select_find_boot_vga_vid_did()
377 uint32_t device_count) in device_select_find_boot_vga_default()
401 uint32_t device_count) in device_select_find_non_cpu()
416 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()
|
| /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/pytorch/torch/csrc/api/src/ |
| D | xpu.cpp | 6 size_t device_count() { in device_count() function
|
| D | cuda.cpp | 12 size_t device_count() { in device_count() function
|
| /external/pytorch/torch/_dynamo/ |
| 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
|
| /external/pytorch/test/distributed/_shard/sharding_plan/ |
| D | test_sharding_plan.py | 38 def __init__(self, chunk_dim=0, device_count=0): argument
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| D | oneDNNContext.h | 38 int device_count = (int)c10::xpu::device_count(); in GpuEngineManager() local
|
| /external/pytorch/aten/src/ATen/xpu/detail/ |
| D | XPUHooks.cpp | 14 const auto device_count = c10::xpu::device_count_ensure_non_zero(); in initXPU() local
|
| /external/pytorch/torch/mps/ |
| D | __init__.py | 26 def device_count() -> int: function
|
| /external/tensorflow/tensorflow/core/util/autotune_maps/ |
| D | autotune_maps_utils.cc | 62 int device_count = GpuDriver::GetDeviceCount(); in GetDeviceIdToIdentifierMap() 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/compiler/xla/service/ |
| D | backend.h | 100 int device_count() const { return stream_executors_.size(); } in device_count() function
|
| /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
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | platform.cc | 109 int device_count = VisibleDeviceCount(); in GetPeerAccessMap() local
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 73 const auto& device_count = config.device_count(); in MaybeCreateNcclCommunicator() local
|
| /external/mesa3d/src/tool/pps/ |
| D | pps_device.cc | 22 uint32_t DrmDevice::device_count() in device_count() function in pps::DrmDevice
|
| /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/coreboot/src/drivers/gfx/generic/ |
| D | chip.h | 66 int device_count; member
|