Searched refs:allocator_parts (Results 1 – 1 of 1) sorted by relevance
93 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GetGPUAllocator() local94 if (allocator_parts.allocator == nullptr) { in GetGPUAllocator()147 allocator_parts = {std::unique_ptr<Allocator>(gpu_allocator), in GetGPUAllocator()153 return allocator_parts.recording_allocator.get(); in GetGPUAllocator()155 return allocator_parts.allocator.get(); in GetGPUAllocator()173 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GPUAllocatorCounter() local174 return allocator_parts.counter.get(); in GPUAllocatorCounter()256 AllocatorParts& allocator_parts = gpu_host_allocators_.back(); in GetGpuHostAllocator() local263 allocator_parts.recording_allocator.reset( in GetGpuHostAllocator()265 allocator_parts.allocator.get(), md, in GetGpuHostAllocator()