Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device.h36 return scoped_allocator_mgr_.get(); in GetScopedAllocatorMgr()
49 std::unique_ptr<ScopedAllocatorMgr> scoped_allocator_mgr_; variable
Dthreadpool_device.cc52 scoped_allocator_mgr_(new ScopedAllocatorMgr(name)) { in ThreadPoolDevice()
80 return scoped_allocator_mgr_->GetContainer(step_id)->GetInstance( in GetScopedAllocator()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h103 return scoped_allocator_mgr_.get(); in GetScopedAllocatorMgr()
127 std::unique_ptr<ScopedAllocatorMgr> scoped_allocator_mgr_; variable
Dgpu_device.cc390 scoped_allocator_mgr_(new ScopedAllocatorMgr(name)), in BaseGPUDevice()
1069 return scoped_allocator_mgr_->GetContainer(step_id)->GetInstance( in GetScopedAllocator()