Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dscoped_allocator_mgr.cc22 Status ScopedAllocatorContainer::AddScopedAllocator( in AddScopedAllocator()
47 ScopedAllocatorContainer::SAField(ScopedAllocator::kBackingIndex, sa); in AddScopedAllocator()
53 allocators_[f.scope_id] = ScopedAllocatorContainer::SAField( in AddScopedAllocator()
59 ScopedAllocator* ScopedAllocatorContainer::GetAllocator(int32 scope_id) { in GetAllocator()
73 ScopedAllocatorInstance* ScopedAllocatorContainer::GetInstance(int32 scope_id) { in GetInstance()
86 void ScopedAllocatorContainer::Drop(int32 scope_id, ScopedAllocator* sa) { in Drop()
99 ScopedAllocatorContainer::~ScopedAllocatorContainer() { in ~ScopedAllocatorContainer()
139 ScopedAllocatorContainer* ScopedAllocatorMgr::GetContainer(int64 step_id) { in GetContainer()
141 ScopedAllocatorContainer* sac = nullptr; in GetContainer()
145 sac = new ScopedAllocatorContainer(this, step_id); in GetContainer()
[all …]
Dscoped_allocator_mgr.h31 class ScopedAllocatorContainer : public core::RefCounted {
47 ScopedAllocatorContainer(const ScopedAllocatorMgr* mgr, int64 step_id) in ScopedAllocatorContainer() function
49 ~ScopedAllocatorContainer();
79 ScopedAllocatorContainer* GetContainer(int64 step_id);
105 std::unordered_map<int64, ScopedAllocatorContainer*> per_step_map_
Dscoped_allocator_mgr_test.cc57 ScopedAllocatorContainer* sac = sam_.GetContainer(step_id_); in SaveInstances()
96 ScopedAllocatorContainer* sac_101 = sam_.GetContainer(101); in TEST_F()
98 ScopedAllocatorContainer* sac_201 = sam_.GetContainer(201); in TEST_F()
101 ScopedAllocatorContainer* also_sac_101 = sam_.GetContainer(101); in TEST_F()
148 ScopedAllocatorContainer* sac = sam_.GetContainer(step_id_); in TEST_F()
Dscoped_allocator.h24 class ScopedAllocatorContainer; variable
49 ScopedAllocatorContainer* container);
74 ScopedAllocatorContainer* container_;
Dscoped_allocator.cc24 ScopedAllocatorContainer* container) in ScopedAllocator()
/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops_test.cc106 ScopedAllocatorContainer* sac = sam->GetContainer(step_id); in PrepOp()