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()
46 std::unique_ptr<ScopedAllocatorMgr> scoped_allocator_mgr_; variable
Dthreadpool_device.cc51 scoped_allocator_mgr_(new ScopedAllocatorMgr(name)) { in ThreadPoolDevice()
83 return scoped_allocator_mgr_->GetContainer(step_id)->GetInstance( in GetScopedAllocator()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h116 return scoped_allocator_mgr_.get(); in GetScopedAllocatorMgr()
135 std::unique_ptr<ScopedAllocatorMgr> scoped_allocator_mgr_; variable
Dgpu_device.cc309 scoped_allocator_mgr_(new ScopedAllocatorMgr(name)), in BaseGPUDevice()
967 return scoped_allocator_mgr_->GetContainer(step_id)->GetInstance( in GetScopedAllocator()