Home
last modified time | relevance | path

Searched defs:numa_node (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.cc73 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForGPU() local
190 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator()
304 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostAllocVisitor()
319 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostFreeVisitor()
Dgpu_host_allocator.h29 explicit GpuHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in GpuHostAllocator()
Dpool_allocator_test.cc132 [&alloc_count, &alloc_size](void* ptr, int numa_node, int64 size) { in TEST()
139 [&free_count, &free_size](void* ptr, int numa_node, int64 size) { in TEST()
Dgpu_device_factory.cc131 int numa_node = i % num_numa_nodes; in CreateDevices() local
Dgpu_device.cc646 int numa_node = attributes().locality().numa_node(); in MakeTensorFromProto() local
1211 int numa_node = dev_locality.numa_node(); in CreateGPUDevice() local
1319 int numa_node = desc->numa_node(); in GetDeviceLocalities() local
/external/tensorflow/tensorflow/c/
Denv.h48 int numa_node; member
/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device_factory.cc50 int numa_node = i % num_numa_nodes; in CreateDevices() local
Dprocess_state.cc62 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator()
Dthreadpool_device.cc128 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator()
Dpool_allocator.h151 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator()
Dlocal_device.cc129 int numa_node = attributes.locality().numa_node(); in LocalDevice() local
/external/tensorflow/tensorflow/core/framework/
Ddevice_attributes.proto26 int32 numa_node = 2; field
Dallocator_registry.cc90 SubAllocator* AllocatorFactoryRegistry::GetSubAllocator(int numa_node) { in GetSubAllocator()
Dallocator.cc80 Allocator* cpu_allocator(int numa_node) { in cpu_allocator()
Dcpu_allocator_impl.cc150 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator()
/external/linux-kselftest/tools/testing/selftests/bpf/
Dbpf_legacy.h16 unsigned int numa_node; member
/external/tensorflow/tensorflow/core/platform/default/
Dport.cc85 int MaxParallelism(int numa_node) { in MaxParallelism()
253 hwloc_obj_t numa_node = GetHWLocTypeIndex(HWLOC_OBJ_NUMANODE, node); in NUMAMalloc() local
/external/tensorflow/tensorflow/core/platform/windows/
Dport.cc60 int MaxParallelism(int numa_node) { in MaxParallelism()
/external/tensorflow/tensorflow/core/kernels/
Dconstant_op.cc282 int numa_node = ctx->device()->NumaNode(); in Compute() local
Dcwise_ops_common.h334 int numa_node = ctx->device()->NumaNode(); in Compute() local
/external/tensorflow/tensorflow/core/platform/
Denv.h452 int numa_node = port::kNUMANoAffinity; member
/external/kernel-headers/original/uapi/rdma/hfi/
Dhfi1_ioctl.h90 __u16 numa_node; /* NUMA node of the assigned device */ member
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.h65 int numa_node() const { return numa_node_; } in numa_node() function
/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc48 int numa_node, uint64 incarnation) { in NewDevice()
/external/strace/
Dbpf_attr.h65 uint32_t numa_node; member

12