ONE - On-device Neural Engine
|
#include <MemoryManager.h>
Public Member Functions | |
TrainableMemoryManager (uint32_t optimizer_vars_count) | |
virtual | ~TrainableMemoryManager ()=default |
void | allocate (void) |
uint8_t * | getOptVarBuffer (const ir::OperandIndex &ind, uint32_t pos_var) const |
Public Member Functions inherited from onert::backend::basic::MemoryManager | |
MemoryManager () | |
MemoryManager (const std::string) | |
virtual | ~MemoryManager ()=default |
void | allocate (void) |
uint8_t * | getBuffer (const ir::OperandIndex &ind) const |
void | deallocate (void) |
void | claimPlan (const ir::OperandIndex &ind, uint32_t size) |
void | releasePlan (const ir::OperandIndex &ind) |
Additional Inherited Members | |
Protected Attributes inherited from onert::backend::basic::MemoryManager | |
std::unordered_map< ir::OperandIndex, Block > | _tensor_mem_map |
std::shared_ptr< IMemoryPlanner< ir::OperandIndex > > | _mem_planner |
std::shared_ptr< Allocator > | _mem_alloc |
Definition at line 34 of file MemoryManager.h.
onert::backend::train::TrainableMemoryManager::TrainableMemoryManager | ( | uint32_t | optimizer_vars_count | ) |
Definition at line 32 of file MemoryManager.cc.
|
virtualdefault |
void onert::backend::train::TrainableMemoryManager::allocate | ( | void | ) |
Definition at line 38 of file MemoryManager.cc.
References onert::backend::basic::MemoryManager::_mem_alloc, and onert::backend::basic::MemoryManager::_mem_planner.
uint8_t * onert::backend::train::TrainableMemoryManager::getOptVarBuffer | ( | const ir::OperandIndex & | ind, |
uint32_t | pos_var | ||
) | const |
Definition at line 47 of file MemoryManager.cc.
References onert::backend::basic::MemoryManager::_mem_planner.