Searched refs:numa_node_ (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_host_allocator.h | 34 numa_node_(numa_node) { in DeviceHostAllocator()
50 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
57 VisitFree(ptr, numa_node_, num_bytes); in Free()
66 const int numa_node_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.cc | 265 if (numa_node_ == port::kNUMANoAffinity) { in Alloc()
269 port::NUMAMalloc(numa_node_, num_bytes, static_cast<int>(alignment)); in Alloc()
271 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
278 VisitFree(ptr, numa_node_, num_bytes); in Free()
279 if (numa_node_ == port::kNUMANoAffinity) { in Free()
|
| D | pool_allocator.h | 154 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {} in BasicCPUAllocator()
166 int numa_node_;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_factory.cc | 87 numa_node_(locality.numa_node()) { in GPUCompatibleCPUDevice()
98 return ps->GetGpuHostAllocator(numa_node_); in GetAllocator()
107 int numa_node_ = port::kNUMANoAffinity; member in tensorflow::GPUCompatibleCPUDevice
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.h | 65 int numa_node() const { return numa_node_; } in numa_node()
216 int numa_node_; variable
311 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()
|