onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor > Class Template Reference

#include <AclMemoryManager.h>

Public Member Functions
	AclMemoryManager ()
virtual	~AclMemoryManager ()=default
virtual void	allocate (void)
virtual void	deallocate (void)
virtual void	startLifetime (const ir::OperandIndex &)
virtual void	finishLifetime (const ir::OperandIndex &)
void	buildTensor (const ir::OperandIndex &ind, const ::arm_compute::TensorInfo &info, size_t rank, size_t num_uses)
void	buildSubtensor (std::shared_ptr< T_ITensor > parent_tensor, const ir::OperandIndex &child_ind, const ::arm_compute::TensorShape &shape, const ::arm_compute::Coordinates &coordinates, size_t rank, bool extent_parent)
ir::OperandIndexMap< std::shared_ptr< T_Tensor > > &	tensors (void)
ir::OperandIndexMap< std::shared_ptr< T_SubTensor > > &	subtensors (void)

Detailed Description

template<typename T_ITensor, typename T_Tensor, typename T_SubTensor>
class onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >

Definition at line 31 of file AclMemoryManager.h.

Constructor & Destructor Documentation

AclMemoryManager()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::AclMemoryManager ( )

inline

Definition at line 34 of file AclMemoryManager.h.

  {
    // DO NOTHING
  }

~AclMemoryManager()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

virtual onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::~AclMemoryManager ( )

virtualdefault

Member Function Documentation

allocate()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

virtual void onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::allocate ( void  )

inlinevirtual

Reimplemented in onert::backend::acl_common::AclLinearMemoryManager< T_ITensor, T_Tensor, T_SubTensor, T_MemoryManager, T_PoolManager, T_LifetimeManager, T_Allocator, T_MemoryGroup >.

Definition at line 41 of file AclMemoryManager.h.

  {
    for (const auto &tensor_entry : _tensors)
    {
      auto tensor = tensor_entry.second;
      tensor->allocator()->allocate();
    }
  }

buildSubtensor()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

void onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::buildSubtensor ( std::shared_ptr< T_ITensor >  parent_tensor, const ir::OperandIndex &  child_ind, const ::arm_compute::TensorShape &  shape, const ::arm_compute::Coordinates &  coordinates, size_t  rank, bool  extent_parent  )

inline

Definition at line 73 of file AclMemoryManager.h.

  {
    auto subtensor =
      std::make_shared<T_SubTensor>(parent_tensor.get(), shape, coordinates, rank, extent_parent);
    _subtensors[child_ind] = subtensor;
  }

buildTensor()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

void onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::buildTensor ( const ir::OperandIndex &  ind, const ::arm_compute::TensorInfo &  info, size_t  rank, size_t  num_uses  )

inline

Definition at line 66 of file AclMemoryManager.h.

  {
    auto tensor = std::make_shared<T_Tensor>(info, rank, num_uses);
    _tensors[ind] = tensor;
  }

References info.

deallocate()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

virtual void onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::deallocate ( void  )

inlinevirtual

Reimplemented in onert::backend::acl_common::AclLinearMemoryManager< T_ITensor, T_Tensor, T_SubTensor, T_MemoryManager, T_PoolManager, T_LifetimeManager, T_Allocator, T_MemoryGroup >.

Definition at line 50 of file AclMemoryManager.h.

  {
    for (const auto &tensor_entry : _tensors)
    {
      auto tensor = tensor_entry.second;
      tensor->allocator()->free();
    }
  }

finishLifetime()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

virtual void onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::finishLifetime ( const ir::OperandIndex &  )

inlinevirtual

Reimplemented in onert::backend::acl_common::AclLinearMemoryManager< T_ITensor, T_Tensor, T_SubTensor, T_MemoryManager, T_PoolManager, T_LifetimeManager, T_Allocator, T_MemoryGroup >.

Definition at line 62 of file AclMemoryManager.h.

  { /* DO NOTHING */
  }

startLifetime()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

virtual void onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::startLifetime ( const ir::OperandIndex &  )

inlinevirtual

Reimplemented in onert::backend::acl_common::AclLinearMemoryManager< T_ITensor, T_Tensor, T_SubTensor, T_MemoryManager, T_PoolManager, T_LifetimeManager, T_Allocator, T_MemoryGroup >.

Definition at line 59 of file AclMemoryManager.h.

  { /* DO NOTHING */
  }

subtensors()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

ir::OperandIndexMap< std::shared_ptr< T_SubTensor > > & onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::subtensors ( void  )

inline

Definition at line 85 of file AclMemoryManager.h.

{ return _subtensors; }

tensors()

template<typename T_ITensor , typename T_Tensor , typename T_SubTensor >

ir::OperandIndexMap< std::shared_ptr< T_Tensor > > & onert::backend::acl_common::AclMemoryManager< T_ITensor, T_Tensor, T_SubTensor >::tensors ( void  )

inline

Definition at line 83 of file AclMemoryManager.h.

{ return _tensors; }

Referenced by onert::backend::acl_common::AclLinearMemoryManager< T_ITensor, T_Tensor, T_SubTensor, T_MemoryManager, T_PoolManager, T_LifetimeManager, T_Allocator, T_MemoryGroup >::finishLifetime(), and onert::backend::acl_common::AclLinearMemoryManager< T_ITensor, T_Tensor, T_SubTensor, T_MemoryManager, T_PoolManager, T_LifetimeManager, T_Allocator, T_MemoryGroup >::startLifetime().

---

The documentation for this class was generated from the following file:

• runtime/onert/backend/acl_common/AclMemoryManager.h