ONE - On-device Neural Engine
|
#include <TensorBuilder.h>
Public Member Functions | |
TensorBuilder (const std::shared_ptr< TensorRegistry > &tensor_reg, const ir::OperandIndexMap< ir::OperandIndex > &shared_memory_operand_indexes={}) | |
TensorBuilder (const std::shared_ptr< TensorRegistry > &tensor_reg, const std::string planner_id, const ir::OperandIndexMap< ir::OperandIndex > &shared_memory_operand_indexes={}) | |
void | registerTensorInfo (const ir::OperandIndex &ind, const ir::OperandInfo &info) |
Register tensor information to allocate on CPU backend. | |
void | notifyFirstUse (const ir::OperandIndex &) |
void | notifyLastUse (const ir::OperandIndex &) |
bool | isRegistered (const ir::OperandIndex &) const |
void | allocate (void) |
const ir::OperandIndexMap< ir::OperandIndex > & | getSharedMemoryOperandIndexes () const |
DynamicTensorManager * | dynamicTensorManager (void) |
Definition at line 37 of file TensorBuilder.h.
onert::backend::basic::TensorBuilder::TensorBuilder | ( | const std::shared_ptr< TensorRegistry > & | tensor_reg, |
const ir::OperandIndexMap< ir::OperandIndex > & | shared_memory_operand_indexes = {} |
||
) |
Definition at line 30 of file TensorBuilder.cc.
onert::backend::basic::TensorBuilder::TensorBuilder | ( | const std::shared_ptr< TensorRegistry > & | tensor_reg, |
const std::string | planner_id, | ||
const ir::OperandIndexMap< ir::OperandIndex > & | shared_memory_operand_indexes = {} |
||
) |
Definition at line 41 of file TensorBuilder.cc.
void onert::backend::basic::TensorBuilder::allocate | ( | void | ) |
Definition at line 90 of file TensorBuilder.cc.
|
inline |
Definition at line 61 of file TensorBuilder.h.
const ir::OperandIndexMap< ir::OperandIndex > & onert::backend::basic::TensorBuilder::getSharedMemoryOperandIndexes | ( | ) | const |
Definition at line 92 of file TensorBuilder.cc.
bool onert::backend::basic::TensorBuilder::isRegistered | ( | const ir::OperandIndex & | ind | ) | const |
Definition at line 85 of file TensorBuilder.cc.
void onert::backend::basic::TensorBuilder::notifyFirstUse | ( | const ir::OperandIndex & | ind | ) |
Definition at line 65 of file TensorBuilder.cc.
References size.
void onert::backend::basic::TensorBuilder::notifyLastUse | ( | const ir::OperandIndex & | ind | ) |
Definition at line 77 of file TensorBuilder.cc.
void onert::backend::basic::TensorBuilder::registerTensorInfo | ( | const ir::OperandIndex & | ind, |
const ir::OperandInfo & | info | ||
) |
Register tensor information to allocate on CPU backend.
[in] | ind | Operand index |
[in] | info | Operand information |
Definition at line 51 of file TensorBuilder.cc.
References info.