ONE - On-device Neural Engine
|
Tensor registry class for acl backends. More...
#include <AclTensorRegistry.h>
Public Member Functions | |
AclTensorRegistry (T_AclTensorManager *tensor_mgr) | |
ITensor * | getITensor (const ir::OperandIndex &ind) override |
Returns pointer of ITensor among native and migrant tensors. | |
ITensor * | getNativeITensor (const ir::OperandIndex &ind) override |
Returns pointer of ITensor among native tensors. | |
auto | getAclTensor (const ir::OperandIndex &ind) |
Public Member Functions inherited from onert::backend::ITensorRegistry | |
virtual | ~ITensorRegistry ()=default |
Deconstruct itself. | |
virtual bool | setMigrantTensor (const ir::OperandIndex &, IPortableTensor *) |
Set the Migrant Tensor which are from other backends. | |
Tensor registry class for acl backends.
This is implemented as a wrapper of AclTensorManager.
Definition at line 34 of file AclTensorRegistry.h.
|
inline |
Definition at line 37 of file AclTensorRegistry.h.
|
inline |
Definition at line 43 of file AclTensorRegistry.h.
|
inlineoverridevirtual |
Returns pointer of ITensor among native and migrant tensors.
Native Tensor is a tensor that is managed by this backend Migrant Tensor is a tensor that is imported from another backend
Implements onert::backend::ITensorRegistry.
Definition at line 39 of file AclTensorRegistry.h.
Referenced by onert::backend::acl_common::AclTensorRegistry< T_AclTensorManager >::getNativeITensor().
|
inlineoverridevirtual |
Returns pointer of ITensor among native tensors.
Unlike getITensor
, this function only searches from native tensors
Implements onert::backend::ITensorRegistry.
Definition at line 41 of file AclTensorRegistry.h.
References onert::backend::acl_common::AclTensorRegistry< T_AclTensorManager >::getITensor().