Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.cc194 Allocator* recording_allocator = nullptr; in GetGPUAllocator() local
201 recording_allocator = new internal::RecordingAllocator( in GetGPUAllocator()
207 std::unique_ptr<Allocator>(recording_allocator)}; in GetGPUAllocator()
210 return allocator_parts.recording_allocator.get(); in GetGPUAllocator()
264 gpu_host_allocators_[0].recording_allocator != nullptr) { in GetGpuHostAllocator()
265 return gpu_host_allocators_[0].recording_allocator.get(); in GetGpuHostAllocator()
332 allocator_parts.recording_allocator.reset( in GetGpuHostAllocator()
339 return gpu_host_allocators_[0].recording_allocator.get(); in GetGpuHostAllocator()
Dgpu_process_state.h149 std::unique_ptr<Allocator> recording_allocator; member