ONE - On-device Neural Engine
|
#include <AclTensorBuilder.h>
Public Types | |
using | T_AclTensorManager = AclTensorManager< T_ITensor, T_Tensor, T_SubTensor > |
using | UsesType = cl_common::UsesType |
Public Member Functions | |
AclTensorBuilder (const ir::Operands &operands, T_AclTensorManager *tensor_mgr) | |
void | registerTensorInfo (const ir::OperandIndex &ind, const ir::OperandInfo &info) |
Register tensor information to allocate on ACL-CL backend. | |
void | notifyFirstUse (const ir::OperandIndex &) |
void | notifyLastUse (const ir::OperandIndex &) |
bool | isRegistered (const ir::OperandIndex &) const |
void | prepare (void) |
void | allocate () |
void | postFunctionPrepare () |
T_AclTensorManager * | acl_tensor_manager (void) |
void | setUsesCount (const ir::OperandIndex &index, size_t num_uses) |
void | parent_map (std::unordered_map< ir::OperandIndex, cl_common::ParentInfo > &&parent_map) |
bool | areSubTensorsOf (const ir::OperandIndex &parent, const ir::OperandIndexSequence &seq) |
bool | isSubTensorOf (const ir::OperandIndex &parent, const ir::OperandIndex &child) |
Check child tensor is allocated as subtensor of parent tensor. | |
Definition at line 42 of file AclTensorBuilder.h.
using onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::T_AclTensorManager = AclTensorManager<T_ITensor, T_Tensor, T_SubTensor> |
Definition at line 45 of file AclTensorBuilder.h.
using onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::UsesType = cl_common::UsesType |
Definition at line 47 of file AclTensorBuilder.h.
onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::AclTensorBuilder | ( | const ir::Operands & | operands, |
T_AclTensorManager * | tensor_mgr | ||
) |
Definition at line 129 of file AclTensorBuilder.h.
|
inline |
Definition at line 68 of file AclTensorBuilder.h.
void onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::allocate | ( | void | ) |
Definition at line 178 of file AclTensorBuilder.h.
References onert::backend::cl_common::createLifetimeMap(), and entry().
bool onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::areSubTensorsOf | ( | const ir::OperandIndex & | parent, |
const ir::OperandIndexSequence & | seq | ||
) |
Definition at line 293 of file AclTensorBuilder.h.
bool onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::isRegistered | ( | const ir::OperandIndex & | ind | ) | const |
Definition at line 165 of file AclTensorBuilder.h.
bool onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::isSubTensorOf | ( | const ir::OperandIndex & | parent, |
const ir::OperandIndex & | child | ||
) |
Check child tensor is allocated as subtensor of parent tensor.
[in] | parent | Index of parent |
[in] | child | Index of child |
true
if child is allocated as subtensor of parent, otherwise false
Definition at line 307 of file AclTensorBuilder.h.
void onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::notifyFirstUse | ( | const ir::OperandIndex & | ind | ) |
Definition at line 153 of file AclTensorBuilder.h.
void onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::notifyLastUse | ( | const ir::OperandIndex & | ind | ) |
Definition at line 159 of file AclTensorBuilder.h.
|
inline |
Definition at line 77 of file AclTensorBuilder.h.
References onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::parent_map().
Referenced by onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::parent_map().
void onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::postFunctionPrepare | ( | void | ) |
Definition at line 205 of file AclTensorBuilder.h.
void onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::prepare | ( | void | ) |
Definition at line 172 of file AclTensorBuilder.h.
void onert::backend::acl_common::AclTensorBuilder< T_ITensor, T_Tensor, T_SubTensor >::registerTensorInfo | ( | const ir::OperandIndex & | ind, |
const ir::OperandInfo & | info | ||
) |
Register tensor information to allocate on ACL-CL backend.
[in] | ind | Operand index |
[in] | info | Tensor information |
[in] | layout | Tensor data layout |
Definition at line 137 of file AclTensorBuilder.h.
References info.
|
inline |
Definition at line 70 of file AclTensorBuilder.h.