onert::backend::train::TensorRegistry Class Reference

#include <TensorRegistry.h>

Collaboration diagram for onert::backend::train::TensorRegistry:

Public Member Functions
BackPropTensor *	getDisposableBackPropTensor (const DisposableTensorIndex &index)
void	setDisposableBackPropTensor (const DisposableTensorIndex &index, std::unique_ptr< BackPropTensor > tensor)
const std::unordered_map< DisposableTensorIndex, std::unique_ptr< BackPropTensor > > &	disposable_back_prop_tensors ()
std::shared_ptr< LayerScopeTensor >	getLayerScopeTensor (const LayerScopeTensorIndex &index)
void	setLayerScopeTensor (const LayerScopeTensorIndex &index, std::shared_ptr< LayerScopeTensor > &tensor)
const std::unordered_map< LayerScopeTensorIndex, std::shared_ptr< LayerScopeTensor > > &	layerscope_tensors ()
Public Member Functions inherited from onert::backend::train::PortableTensorRegistryTemplate< Tensor, TrainableTensor, BackPropTensor, GradientTensor >
ITensor *	getITensor (const ir::OperandIndex &index) override
	Returns pointer of ITensor among native and migrant tensors.
ITensor *	getNativeITensor (const ir::OperandIndex &index) override
	Returns pointer of ITensor among native tensors.
ITensor *	getBackPropITensor (const ir::OperandIndex &index) override
	Returns pointer of ITensor for back propatation.
ITensor *	getGradientITensor (const ir::OperandIndex &index) override
	Returns pointer of ITensor for gradient.
void	iterateTrainableTensors (const std::function< void(const ir::OperandIndex &, const train::ITrainableTensor *)> &fn) const override
	Iterate ITrainableTensors with fn.
IPortableTensor *	getPortableTensor (const ir::OperandIndex &index)
Tensor *	getNonConstTensor (const ir::OperandIndex &index)
TrainableTensor *	getTrainableTensor (const ir::OperandIndex &index)
BackPropTensor *	getBackPropTensor (const ir::OperandIndex &index)
GradientTensor *	getGradientTensor (const ir::OperandIndex &index)
TrainingTensors	getTrainingTensors (const ir::OperandIndex &index)
bool	setMigrantTensor (const ir::OperandIndex &index, IPortableTensor *tensor) override
	Set the Migrant Tensor which are from other backends.
void	setNonConstTensor (const ir::OperandIndex &index, std::unique_ptr< Tensor > tensor)
void	setTrainableTensor (const ir::OperandIndex &index, std::unique_ptr< TrainableTensor > tensor)
void	setBackPropTensor (const ir::OperandIndex &index, std::unique_ptr< BackPropTensor > tensor)
void	setGradientTensor (const ir::OperandIndex &index, std::unique_ptr< GradientTensor > tensor)
const ir::OperandIndexMap< std::unique_ptr< TrainableTensor > > &	trainable_tensors ()
const ir::OperandIndexMap< std::unique_ptr< Tensor > > &	nonconst_tensors ()
const ir::OperandIndexMap< std::unique_ptr< Tensor > > &	back_prop_tensors ()
const ir::OperandIndexMap< std::unique_ptr< GradientTensor > > &	gradient_tensors ()
Public Member Functions inherited from onert::backend::ITensorRegistry
virtual	~ITensorRegistry ()=default
	Deconstruct itself.

Additional Inherited Members
Public Types inherited from onert::backend::train::PortableTensorRegistryTemplate< Tensor, TrainableTensor, BackPropTensor, GradientTensor >
using	TrainingTensors = std::tuple< TrainableTensor *, GradientTensor * >

Detailed Description

Definition at line 30 of file TensorRegistry.h.

Member Function Documentation

disposable_back_prop_tensors()

const std::unordered_map< DisposableTensorIndex, std::unique_ptr< BackPropTensor > > & onert::backend::train::TensorRegistry::disposable_back_prop_tensors ( )

inline

Definition at line 56 of file TensorRegistry.h.

  {
    return _disposable_back_prop;
  }

getDisposableBackPropTensor()

BackPropTensor * onert::backend::train::TensorRegistry::getDisposableBackPropTensor ( const DisposableTensorIndex &  index )

inline

Definition at line 34 of file TensorRegistry.h.

  {
    auto itr = _disposable_back_prop.find(index);
    if (itr != _disposable_back_prop.end())
      return itr->second.get();
 
    return nullptr;
  }

getLayerScopeTensor()

std::shared_ptr< LayerScopeTensor > onert::backend::train::TensorRegistry::getLayerScopeTensor ( const LayerScopeTensorIndex &  index )

inline

Definition at line 61 of file TensorRegistry.h.

  {
    auto itr = _layer_scope.find(index);
    if (itr != _layer_scope.end())
      return itr->second;
 
    return nullptr;
  }

layerscope_tensors()

const std::unordered_map< LayerScopeTensorIndex, std::shared_ptr< LayerScopeTensor > > & onert::backend::train::TensorRegistry::layerscope_tensors ( )

inline

Definition at line 83 of file TensorRegistry.h.

  {
    return _layer_scope;
  }

setDisposableBackPropTensor()

void onert::backend::train::TensorRegistry::setDisposableBackPropTensor ( const DisposableTensorIndex &  index, std::unique_ptr< BackPropTensor >  tensor  )

inline

Definition at line 43 of file TensorRegistry.h.

  {
    assert(tensor != nullptr);
    auto itr = _disposable_back_prop.find(index);
    if (itr != _disposable_back_prop.end())
      throw std::runtime_error{
        "Tried to set a disposable tensor but another disposable tensor already exists."};
 
    _disposable_back_prop[index] = std::move(tensor);
  }

setLayerScopeTensor()

void onert::backend::train::TensorRegistry::setLayerScopeTensor ( const LayerScopeTensorIndex &  index, std::shared_ptr< LayerScopeTensor > &  tensor  )

inline

Definition at line 70 of file TensorRegistry.h.

  {
    assert(tensor != nullptr);
    auto itr = _layer_scope.find(index);
    if (itr != _layer_scope.end())
      throw std::runtime_error{
        "Tried to set a layer scope tensor but another layer scope tensor already exists."};
 
    _layer_scope[index] = tensor;
  }

---

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

• runtime/onert/backend/train/TensorRegistry.h