Searched refs:allocator_parts (Results 1 – 1 of 1) sorted by relevance
151 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GetGPUAllocator() local152 if (allocator_parts.allocator == nullptr) { in GetGPUAllocator()204 allocator_parts = {std::unique_ptr<Allocator>(gpu_allocator), in GetGPUAllocator()210 return allocator_parts.recording_allocator.get(); in GetGPUAllocator()212 return allocator_parts.allocator.get(); in GetGPUAllocator()234 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GPUAllocatorCounter() local235 if (allocator_parts.counter.get() == nullptr) { in GPUAllocatorCounter()237 allocator_parts.bfc_allocator->SetTimingCounter(timing_counter); in GPUAllocatorCounter()238 allocator_parts.counter.reset(timing_counter); in GPUAllocatorCounter()240 return allocator_parts.counter.get(); in GPUAllocatorCounter()[all …]