|
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 30 of file AclTensorRegistry.h.
|
inline |
Definition at line 33 of file AclTensorRegistry.h.
|
inline |
Definition at line 39 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 35 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 37 of file AclTensorRegistry.h.
References onert::backend::acl_common::AclTensorRegistry< T_AclTensorManager >::getITensor().