Searched refs:GetCPUAllocator (Results 1 – 11 of 11) sorted by relevance
61 ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateDevices()65 ProcessState::singleton()->GetCPUAllocator(port::kNUMANoAffinity)); in CreateDevices()
62 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator() function in tensorflow::ProcessState
68 Allocator* GetCPUAllocator(int numa_node) override;
133 ProcessState::singleton()->GetCPUAllocator(numa_node); in LocalDevice()
46 syclInterface->GetSYCLAllocator(i), syclInterface->GetCPUAllocator(i), in CreateDevices()
147 Allocator* GetCPUAllocator(size_t i = 0) const {
88 return ps->GetCPUAllocator(numa_node); in cpu_allocator()
52 virtual Allocator* GetCPUAllocator(int numa_node) = 0;
136 ProcessState::singleton()->GetCPUAllocator(numa_node))); in CreateDevices()
194 return process_state_->GetCPUAllocator(numa_node); in GetGpuHostAllocator()
1243 gpu_allocator, ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateGPUDevice()