ONE - On-device Neural Engine
|
#include <TensorBuilder.h>
Public Member Functions | |
TensorBuilder (const std::shared_ptr< TensorRegistry > &tensor_reg) | |
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) |
DynamicTensorManager * | dynamicTensorManager (void) |
basic::Tensor * | nativeOwnTensorAt (const ir::OperandIndex &ind) |
Get tensor with a specific OperandIndex. | |
Definition at line 37 of file TensorBuilder.h.
onert::backend::builtin::TensorBuilder::TensorBuilder | ( | const std::shared_ptr< TensorRegistry > & | tensor_reg | ) |
Definition at line 30 of file TensorBuilder.cc.
void onert::backend::builtin::TensorBuilder::allocate | ( | void | ) |
Definition at line 90 of file TensorBuilder.cc.
DynamicTensorManager * onert::backend::builtin::TensorBuilder::dynamicTensorManager | ( | void | ) |
Definition at line 92 of file TensorBuilder.cc.
bool onert::backend::builtin::TensorBuilder::isRegistered | ( | const ir::OperandIndex & | ind | ) | const |
Definition at line 80 of file TensorBuilder.cc.
basic::Tensor * onert::backend::builtin::TensorBuilder::nativeOwnTensorAt | ( | const ir::OperandIndex & | ind | ) |
Get tensor with a specific OperandIndex.
ind | OperandIndex for the tensor. There must exist a tensor with this ind. If not, program will crash with assert or exception. |
Definition at line 97 of file TensorBuilder.cc.
void onert::backend::builtin::TensorBuilder::notifyFirstUse | ( | const ir::OperandIndex & | ind | ) |
Definition at line 53 of file TensorBuilder.cc.
References size.
void onert::backend::builtin::TensorBuilder::notifyLastUse | ( | const ir::OperandIndex & | ind | ) |
Definition at line 68 of file TensorBuilder.cc.
void onert::backend::builtin::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 38 of file TensorBuilder.cc.