ONE - On-device Neural Engine
|
Class to plan memory by bump way. More...
#include <MemoryPlanner.h>
Public Member Functions | |
void | claim (const Index &, size_t) override |
Claim memory for tensor by bump way. | |
void | release (const Index &) override |
Release memory for tensor by bump way. | |
uint32_t | capacity () override |
Get capacity for memory planning. | |
MemoryPlans & | memory_plans () override |
Get MemoryPlans. | |
Public Member Functions inherited from onert::backend::basic::IMemoryPlanner< Index > | |
virtual | ~IMemoryPlanner ()=default |
Additional Inherited Members | |
Public Types inherited from onert::backend::basic::IMemoryPlanner< Index > | |
using | MemoryPlans = std::unordered_map< Index, Block > |
Class to plan memory by bump way.
Definition at line 44 of file MemoryPlanner.h.
|
inlineoverridevirtual |
Get capacity for memory planning.
Implements onert::backend::basic::IMemoryPlanner< Index >.
Definition at line 65 of file MemoryPlanner.h.
|
overridevirtual |
Claim memory for tensor by bump way.
[in] | index | The tensor index |
[in] | size | The size of the memory |
Implements onert::backend::basic::IMemoryPlanner< Index >.
Definition at line 33 of file MemoryPlanner.cc.
References onert::backend::basic::Block::offset, size, and VERBOSE.
|
inlineoverridevirtual |
Get MemoryPlans.
Implements onert::backend::basic::IMemoryPlanner< Index >.
Definition at line 70 of file MemoryPlanner.h.
|
overridevirtual |
Release memory for tensor by bump way.
[in] | index | The tensor index |
Implements onert::backend::basic::IMemoryPlanner< Index >.
Definition at line 42 of file MemoryPlanner.cc.
References VERBOSE.