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.cc137 Allocator* recording_allocator = nullptr; in GetGPUAllocator() local
144 recording_allocator = new internal::RecordingAllocator( in GetGPUAllocator()
150 std::unique_ptr<Allocator>(recording_allocator)}; in GetGPUAllocator()
153 return allocator_parts.recording_allocator.get(); in GetGPUAllocator()
198 gpu_host_allocators_[0].recording_allocator != nullptr) { in GetGpuHostAllocator()
199 return gpu_host_allocators_[0].recording_allocator.get(); in GetGpuHostAllocator()
263 allocator_parts.recording_allocator.reset( in GetGpuHostAllocator()
270 return gpu_host_allocators_[0].recording_allocator.get(); in GetGpuHostAllocator()
Dgpu_process_state.h141 std::unique_ptr<Allocator> recording_allocator; member