| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_process_state.cc | 91 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForGPU() local
291 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator()
409 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostAllocVisitor()
424 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostFreeVisitor()
|
| H A D | pool_allocator_test.cc | 135 [&alloc_count, &alloc_size](void* ptr, int numa_node, int64_t size) { in TEST()
142 [&free_count, &free_size](void* ptr, int numa_node, int64_t size) { in TEST()
|
| H A D | gpu_device_factory.cc | 131 int numa_node = i % num_numa_nodes; in CreateDevices() local
|
| H A D | gpu_device.cc | 842 int numa_node = attributes().locality().numa_node(); in MakeTensorFromProto() local
1572 int numa_node = dev_locality.numa_node(); in CreateGPUDevice() local
1692 int numa_node = desc->numa_node(); in GetDeviceLocalities() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_process_state.cc | 74 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForPluggableDevice() local
141 int numa_node) { in GetPluggableDeviceHostAllocator()
|
| H A D | pluggable_device_factory.cc | 240 int numa_node = dev_locality.numa_node(); in CreatePluggableDevice() local
298 int numa_node = desc->numa_node(); in GetDeviceLocalities() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tsl/platform/default/ |
| H A D | port.cc | 119 int MaxParallelism(int numa_node) { in MaxParallelism()
287 hwloc_obj_t numa_node = GetHWLocTypeIndex(HWLOC_OBJ_NUMANODE, node); in NUMAMalloc() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | env.h | 49 int numa_node; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | threadpool_device_factory.cc | 50 int numa_node = i % num_numa_nodes; in CreateDevices() local
|
| H A D | process_state.cc | 64 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator()
|
| H A D | pool_allocator.h | 156 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator()
|
| H A D | local_device.cc | 129 int numa_node = attributes.locality().numa_node(); in LocalDevice() local
|
| H A D | threadpool_device.cc | 262 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/ |
| H A D | device_host_allocator.h | 29 explicit DeviceHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in DeviceHostAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_attributes.proto | 27 int32 numa_node = 2; field
|
| H A D | allocator.cc | 79 Allocator* cpu_allocator(int numa_node) { in cpu_allocator()
|
| H A D | allocator_registry.cc | 90 SubAllocator* AllocatorFactoryRegistry::GetSubAllocator(int numa_node) { in GetSubAllocator()
|
| H A D | cpu_allocator_impl.cc | 186 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator()
|
| /aosp_15_r20/external/pytorch/c10/util/ |
| H A D | numa.cpp | 49 int numa_node = -1; in GetNUMANode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | constant_op.cc | 253 int numa_node = ctx->device()->NumaNode(); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tsl/platform/windows/ |
| H A D | port.cc | 70 int MaxParallelism(int numa_node) { in MaxParallelism()
|
| /aosp_15_r20/prebuilts/runtime/mainline/runtime/sdk/android/arm64/include/bionic/libc/kernel/uapi/rdma/hfi/ |
| H A D | hfi1_ioctl.h | 26 __u16 numa_node; member
|
| /aosp_15_r20/prebuilts/build-tools/sysroots/aarch64-unknown-linux-musl/include/rdma/hfi/ |
| H A D | hfi1_ioctl.h | 26 __u16 numa_node; member
|
| /aosp_15_r20/prebuilts/vndk/v33/x86/include/bionic/libc/kernel/uapi/rdma/hfi/ |
| D | hfi1_ioctl.h | 38 __u16 numa_node; member
|
| /aosp_15_r20/prebuilts/vndk/v31/x86/include/bionic/libc/kernel/uapi/rdma/hfi/ |
| D | hfi1_ioctl.h | 38 __u16 numa_node; member
|