ONE - On-device Neural Engine
|
Class to plan memory by Weighted Interval Color algorithm. More...
#include <MemoryPlanner.h>
Public Member Functions | |
WICPlanner () | |
void | claim (const Index &, size_t) override |
Claim memory for tensor by WIC algorithm. | |
void | release (const Index &) override |
Release memory for tensor by WIC algorithm. | |
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 Weighted Interval Color algorithm.
Definition at line 118 of file MemoryPlanner.h.
onert::backend::train::WICPlanner< Index >::WICPlanner | ( | ) |
Definition at line 114 of file MemoryPlanner.cc.
|
inlineoverridevirtual |
Get capacity for memory planning.
Implements onert::backend::basic::IMemoryPlanner< Index >.
Definition at line 141 of file MemoryPlanner.h.
|
overridevirtual |
Claim memory for tensor by WIC algorithm.
[in] | index | The tensor index |
[in] | size | The size of the memory |
Implements onert::backend::basic::IMemoryPlanner< Index >.
Definition at line 121 of file MemoryPlanner.cc.
|
overridevirtual |
Get MemoryPlans.
Implements onert::backend::basic::IMemoryPlanner< Index >.
Definition at line 202 of file MemoryPlanner.cc.
|
overridevirtual |
Release memory for tensor by WIC algorithm.
[in] | index | The tensor index |
Implements onert::backend::basic::IMemoryPlanner< Index >.
Definition at line 135 of file MemoryPlanner.cc.
References VERBOSE.