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