onert::backend::train::LayerScopeMemoryManager Class Reference

#include <MemoryManager.h>

Public Member Functions
	LayerScopeMemoryManager ()
void	allocate (void)
uint8_t *	getBuffer (const LayerScopeTensorIndex &ind) const
void	deallocate (void)
void	claimPlan (const LayerScopeTensorIndex &ind, uint32_t size)
void	releasePlan (const LayerScopeTensorIndex &ind)

Detailed Description

Definition at line 71 of file MemoryManager.h.

Constructor & Destructor Documentation

LayerScopeMemoryManager()

onert::backend::train::LayerScopeMemoryManager::LayerScopeMemoryManager

(

)

Definition at line 96 of file MemoryManager.cc.

                                                 : _mem_planner{createMemoryPlanner()}
{
  // DO NOTHING
}

Member Function Documentation

allocate()

void onert::backend::train::LayerScopeMemoryManager::allocate

(

void

)

Definition at line 107 of file MemoryManager.cc.

{
  _mem_alloc = std::make_shared<basic::Allocator>(_mem_planner->capacity());
  assert(_mem_alloc->base());
}

claimPlan()

void onert::backend::train::LayerScopeMemoryManager::claimPlan	(	const LayerScopeTensorIndex &	ind,
		uint32_t	size
	)

Definition at line 122 of file MemoryManager.cc.

{
  _mem_planner->claim(ind, size);
}

References size.

deallocate()

void onert::backend::train::LayerScopeMemoryManager::deallocate

(

void

)

Definition at line 120 of file MemoryManager.cc.

{ _mem_alloc->release(); }

getBuffer()

uint8_t * onert::backend::train::LayerScopeMemoryManager::getBuffer

(

const LayerScopeTensorIndex &

ind

)

const

Definition at line 113 of file MemoryManager.cc.

{
  assert(_mem_planner->memory_plans().find(ind) != _mem_planner->memory_plans().end());
  const auto &mem_blk = _mem_planner->memory_plans().at(ind);
  return _mem_alloc->base() + mem_blk.offset;
}

releasePlan()

void onert::backend::train::LayerScopeMemoryManager::releasePlan

(

const LayerScopeTensorIndex &

ind

)

Definition at line 127 of file MemoryManager.cc.

{
  _mem_planner->release(ind);
}

---

The documentation for this class was generated from the following files:

• runtime/onert/backend/train/MemoryManager.h
• runtime/onert/backend/train/MemoryManager.cc