| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| 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()
|
| 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()
|
| D | gpu_device_factory.cc | 131 int numa_node = i % num_numa_nodes; in CreateDevices() local
|
| 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
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_process_state.cc | 74 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForPluggableDevice() local
141 int numa_node) { in GetPluggableDeviceHostAllocator()
|
| 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
|
| D | pluggable_device.cc | 377 int numa_node = attributes().locality().numa_node(); in MakeTensorFromProto() local
|
| /external/tensorflow/tensorflow/c/ |
| D | env.h | 49 int numa_node; member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device_factory.cc | 50 int numa_node = i % num_numa_nodes; in CreateDevices() local
|
| D | process_state.cc | 64 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator()
|
| D | pool_allocator.h | 156 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator()
|
| D | local_device.cc | 129 int numa_node = attributes.locality().numa_node(); in LocalDevice() local
|
| D | threadpool_device.cc | 262 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_host_allocator.h | 29 explicit DeviceHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in DeviceHostAllocator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_attributes.proto | 27 int32 numa_node = 2; field
|
| D | allocator.cc | 79 Allocator* cpu_allocator(int numa_node) { in cpu_allocator()
|
| D | allocator_registry.cc | 90 SubAllocator* AllocatorFactoryRegistry::GetSubAllocator(int numa_node) { in GetSubAllocator()
|
| D | cpu_allocator_impl.cc | 186 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator()
|
| /external/tensorflow/tensorflow/tsl/platform/default/ |
| 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
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op.cc | 254 int numa_node = ctx->device()->NumaNode(); in Compute() local
|
| D | cwise_ops_common.h | 343 int numa_node = ctx->device()->NumaNode(); in Compute() local
|
| /external/tensorflow/tensorflow/tsl/platform/windows/ |
| D | port.cc | 70 int MaxParallelism(int numa_node) { in MaxParallelism()
|
| /external/kernel-headers/original/uapi/rdma/hfi/ |
| D | hfi1_ioctl.h | 90 __u16 numa_node; /* NUMA node of the assigned device */ member
|
| /external/libbpf/src/ |
| D | bpf.h | 52 __u32 numa_node; member
71 __u32 numa_node; member
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | device_description.h | 202 int numa_node() const { return numa_node_; } in numa_node() function
|