HomeSort by relevance Sort by last modified time
    Searched defs:numa_node_ (Results 1 - 5 of 5) sorted by null

  /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.h 153 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {}
162 int numa_node_; member in class:tensorflow::BasicCPUAllocator
  /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_;
  /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; }

Completed in 315 milliseconds