Home
last modified time | relevance | path

Searched refs:allocator_parts (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.cc93 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GetGPUAllocator() local
94 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() local
174 return allocator_parts.counter.get(); in GPUAllocatorCounter()
256 AllocatorParts& allocator_parts = gpu_host_allocators_.back(); in GetGpuHostAllocator() local
263 allocator_parts.recording_allocator.reset( in GetGpuHostAllocator()
265 allocator_parts.allocator.get(), md, in GetGpuHostAllocator()