onert::backend::basic::TensorBuilder Class Reference

#include <TensorBuilder.h>

Public Member Functions
	TensorBuilder (const std::shared_ptr< TensorRegistry > &tensor_reg, const ir::OperandIndexMap< ir::OperandIndex > &shared_memory_operand_indexes={})
	TensorBuilder (const std::shared_ptr< TensorRegistry > &tensor_reg, const std::string planner_id, const ir::OperandIndexMap< ir::OperandIndex > &shared_memory_operand_indexes={})
void	registerTensorInfo (const ir::OperandIndex &ind, const ir::OperandInfo &info)
	Register tensor information to allocate on CPU backend.
void	notifyFirstUse (const ir::OperandIndex &)
void	notifyLastUse (const ir::OperandIndex &)
bool	isRegistered (const ir::OperandIndex &) const
void	allocate (void)
const ir::OperandIndexMap< ir::OperandIndex > &	getSharedMemoryOperandIndexes () const
DynamicTensorManager *	dynamicTensorManager (void)

Detailed Description

Definition at line 33 of file TensorBuilder.h.

Constructor & Destructor Documentation

TensorBuilder() [1/2]

onert::backend::basic::TensorBuilder::TensorBuilder	(	const std::shared_ptr< TensorRegistry > &	tensor_reg,
		const ir::OperandIndexMap< ir::OperandIndex > &	shared_memory_operand_indexes = {}
	)

Definition at line 26 of file TensorBuilder.cc.

  : _tensor_reg{tensor_reg}, _dynamic_tensor_mgr{new DynamicTensorManager(_tensor_reg)},
    _static_tensor_mgr{new StaticTensorManager(_tensor_reg, _dynamic_tensor_mgr.get(),
                                               shared_memory_operand_indexes)},
    _shared_memory_operand_indexes{shared_memory_operand_indexes}
{
  /* empty */
}

TensorBuilder() [2/2]

onert::backend::basic::TensorBuilder::TensorBuilder	(	const std::shared_ptr< TensorRegistry > &	tensor_reg,
		const std::string	planner_id,
		const ir::OperandIndexMap< ir::OperandIndex > &	shared_memory_operand_indexes = {}
	)

Definition at line 37 of file TensorBuilder.cc.

  : _tensor_reg{tensor_reg}, _dynamic_tensor_mgr{new DynamicTensorManager(_tensor_reg)},
    _static_tensor_mgr{new StaticTensorManager(_tensor_reg, planner_id, _dynamic_tensor_mgr.get(),
                                               shared_memory_operand_indexes)}
{
  /* empty */
}

Member Function Documentation

allocate()

void onert::backend::basic::TensorBuilder::allocate

(

void

)

Definition at line 86 of file TensorBuilder.cc.

{ _static_tensor_mgr->allocateNonconsts(); }

dynamicTensorManager()

DynamicTensorManager * onert::backend::basic::TensorBuilder::dynamicTensorManager ( void  )

inline

Definition at line 57 of file TensorBuilder.h.

{ return _dynamic_tensor_mgr.get(); }

getSharedMemoryOperandIndexes()

const ir::OperandIndexMap< ir::OperandIndex > & onert::backend::basic::TensorBuilder::getSharedMemoryOperandIndexes

(

)

const

Definition at line 88 of file TensorBuilder.cc.

{
  return _shared_memory_operand_indexes;
}

isRegistered()

bool onert::backend::basic::TensorBuilder::isRegistered

(

const ir::OperandIndex &

ind

)

const

Definition at line 81 of file TensorBuilder.cc.

{
  return _tensor_info_map.find(ind) != _tensor_info_map.end();
}

notifyFirstUse()

void onert::backend::basic::TensorBuilder::notifyFirstUse

(

const ir::OperandIndex &

ind

)

Definition at line 61 of file TensorBuilder.cc.

{
  assert(_tensor_info_map.find(ind) != _tensor_info_map.end());
  const auto &tensor_info = _tensor_info_map.at(ind);
 
  if (!_tensor_reg->getNativeTensor(ind)->is_dynamic())
  {
    const auto size = tensor_info.total_size();
    _static_tensor_mgr->claimPlan(ind, size);
  }
}

References size.

notifyLastUse()

void onert::backend::basic::TensorBuilder::notifyLastUse

(

const ir::OperandIndex &

ind

)

Definition at line 73 of file TensorBuilder.cc.

{
  if (!_tensor_reg->getNativeTensor(ind)->is_dynamic())
  {
    _static_tensor_mgr->releasePlan(ind);
  }
}

registerTensorInfo()

void onert::backend::basic::TensorBuilder::registerTensorInfo	(	const ir::OperandIndex &	ind,
		const ir::OperandInfo &	info
	)

Register tensor information to allocate on CPU backend.

Parameters

[in]	ind	Operand index
[in]	info	Operand information

Definition at line 47 of file TensorBuilder.cc.

{
  _tensor_info_map.emplace(ind, info);
 
  if (info.isDynamic())
  {
    _dynamic_tensor_mgr->buildTensor(ind, info);
  }
  else
  {
    _static_tensor_mgr->buildTensor(ind, info, info.isConstant());
  }
}

References info.

---

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

• runtime/onert/core/include/backend/basic/TensorBuilder.h
• runtime/onert/core/src/backend/basic/TensorBuilder.cc