Searched refs:numa_node_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_host_allocator.h | 34 numa_node_(numa_node) { in GpuHostAllocator()
48 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
55 VisitFree(ptr, numa_node_, num_bytes); in Free()
62 const int numa_node_; variable
|
| D | gpu_device_factory.cc | 86 numa_node_(locality.numa_node()) { in GPUCompatibleCPUDevice()
97 return ps->GetGpuHostAllocator(numa_node_); in GetAllocator()
106 int numa_node_ = port::kNUMANoAffinity; member in tensorflow::GPUCompatibleCPUDevice
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.cc | 262 if (numa_node_ == port::kNUMANoAffinity) { in Alloc()
266 port::NUMAMalloc(numa_node_, num_bytes, static_cast<int>(alignment)); in Alloc()
268 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
275 VisitFree(ptr, numa_node_, num_bytes); in Free()
276 if (numa_node_ == port::kNUMANoAffinity) { in Free()
|
| D | pool_allocator.h | 153 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {} in BasicCPUAllocator()
162 int numa_node_;
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.h | 65 int numa_node() const { return numa_node_; } in numa_node()
212 int numa_node_; variable
305 void set_numa_node(int value) { device_description_->numa_node_ = value; } in set_numa_node()
|
| D | device_description.cc | 55 numa_node_(-1), in DeviceDescription()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 151 int numa_node_; member in tensorflow::__anon0ef87b4d0111::GdrMemoryManager
234 numa_node_ = TryToReadNumaNode(listening_->verbs->device); in Init()
264 GPUProcessState::singleton()->AddGPUAllocVisitor(numa_node_, in Init()
266 LOG(INFO) << "Instrumenting GPU allocator for NUMA " << numa_node_; in Init()
|