ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
onert::backend::basic::DynamicTensorManager Class Reference

Class to manage dynamic tensor and its memory. More...

#include <DynamicTensorManager.h>

Public Member Functions

 DynamicTensorManager (const std::shared_ptr< TensorRegistry > &reg)
 
virtual ~DynamicTensorManager ()=default
 
void buildTensor (const ir::OperandIndex &ind, const ir::OperandInfo &tensor_info)
 
std::shared_ptr< DynamicMemoryManagerdynamic_mem_mgr ()
 

Detailed Description

Class to manage dynamic tensor and its memory.

Definition at line 37 of file DynamicTensorManager.h.

Constructor & Destructor Documentation

◆ DynamicTensorManager()

onert::backend::basic::DynamicTensorManager::DynamicTensorManager ( const std::shared_ptr< TensorRegistry > &  reg)

Definition at line 25 of file DynamicTensorManager.cc.

26 : _dynamic_mem_mgr{new DynamicMemoryManager()}, _tensors{reg}
27{
28 // DO NOTHING
29}

◆ ~DynamicTensorManager()

virtual onert::backend::basic::DynamicTensorManager::~DynamicTensorManager ( )
virtualdefault

Member Function Documentation

◆ buildTensor()

void onert::backend::basic::DynamicTensorManager::buildTensor ( const ir::OperandIndex ind,
const ir::OperandInfo tensor_info 
)

Definition at line 31 of file DynamicTensorManager.cc.

33{
34 assert(_tensors->getNativeTensor(ind) == nullptr);
35 auto tensor = std::make_unique<Tensor>(tensor_info, _dynamic_mem_mgr.get());
36 _tensors->setNativeTensor(ind, std::move(tensor));
37}

◆ dynamic_mem_mgr()

std::shared_ptr< DynamicMemoryManager > onert::backend::basic::DynamicTensorManager::dynamic_mem_mgr ( )
inline

Definition at line 46 of file DynamicTensorManager.h.

46{ return _dynamic_mem_mgr; }

Referenced by onert::backend::basic::StaticTensorManager::buildTensor().


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