Searched refs:GetCPUAllocator (Results 1 – 9 of 9) sorted by relevance
61 ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateDevices()65 ProcessState::singleton()->GetCPUAllocator(port::kNUMANoAffinity)); in CreateDevices()
68 Allocator* GetCPUAllocator(int numa_node) override;
133 ProcessState::singleton()->GetCPUAllocator(numa_node); in LocalDevice()
64 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator() function in tensorflow::ProcessState
89 return ps->GetCPUAllocator(numa_node); in cpu_allocator()
52 virtual Allocator* GetCPUAllocator(int numa_node) = 0;
136 ProcessState::singleton()->GetCPUAllocator(numa_node))); in CreateDevices()
250 return process_state_->GetCPUAllocator(numa_node); in GetGpuHostAllocator()
1455 gpu_allocator, ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateGPUDevice()