Searched refs:GetCPUAllocator (Results 1 – 11 of 11) sorted by relevance
55 ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateDevices()59 ProcessState::singleton()->GetCPUAllocator(port::kNUMANoAffinity)); in CreateDevices()
68 Allocator* GetCPUAllocator(int numa_node);
62 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator() function in tensorflow::ProcessState
140 ProcessState::singleton()->GetCPUAllocator(numa_node); in LocalDevice()
42 syclInterface->GetSYCLAllocator(i), syclInterface->GetCPUAllocator(i), in CreateDevices()
147 Allocator* GetCPUAllocator(size_t i = 0) const {
51 virtual Allocator* GetCPUAllocator(int numa_node) = 0;
232 return ps->GetCPUAllocator(numa_node); in cpu_allocator()
129 ProcessState::singleton()->GetCPUAllocator(numa_node))); in CreateDevices()
184 return process_state_->GetCPUAllocator(numa_node); in GetGpuHostAllocator()
1221 gpu_allocator, ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateGPUDevice()