onert::compiler::train::TensorRegistries Class Reference

#include <TensorRegistries.h>

Public Member Functions
	TensorRegistries ()=default
	TensorRegistries (const backend::train::TrainableBackendContexts &backend_contexts, bool include_builtin)
std::unordered_set< std::shared_ptr< backend::train::ITensorRegistry > >::const_iterator	begin () const
std::unordered_set< std::shared_ptr< backend::train::ITensorRegistry > >::const_iterator	end () const
std::shared_ptr< backend::builtin::train::TensorRegistry >	getBuiltinTensorRegistry () const
backend::ITensor *	getITensor (ir::OperandIndex index) const
backend::ITensor *	getBackPropITensor (ir::OperandIndex index) const
void	iterateTrainableTensors (const std::function< void(const ir::OperandIndex &, const backend::train::ITrainableTensor *)> &fn) const

Detailed Description

Definition at line 36 of file TensorRegistries.h.

Constructor & Destructor Documentation

TensorRegistries() [1/2]

onert::compiler::train::TensorRegistries::TensorRegistries ( )

default

TensorRegistries() [2/2]

onert::compiler::train::TensorRegistries::TensorRegistries ( const backend::train::TrainableBackendContexts &  backend_contexts, bool  include_builtin  )

inline

Definition at line 41 of file TensorRegistries.h.

  {
    for (const auto &e : backend_contexts)
    {
      auto tensor_reg = e.second->tensor_registry();
      if (e.first->config()->id() == backend::builtin::Config::ID)
      {
        _builtin_tensor_reg =
          std::dynamic_pointer_cast<backend::builtin::train::TensorRegistry>(tensor_reg);
        if (include_builtin)
          _tensor_regs.insert(tensor_reg);
      }
      else
      {
        _tensor_regs.insert(tensor_reg);
      }
    }
  }

References onert::backend::builtin::Config::ID.

Member Function Documentation

begin()

std::unordered_set< std::shared_ptr< backend::train::ITensorRegistry > >::const_iterator onert::compiler::train::TensorRegistries::begin ( ) const

inline

Definition at line 61 of file TensorRegistries.h.

  {
    return _tensor_regs.cbegin();
  }

end()

std::unordered_set< std::shared_ptr< backend::train::ITensorRegistry > >::const_iterator onert::compiler::train::TensorRegistries::end ( ) const

inline

Definition at line 65 of file TensorRegistries.h.

  {
    return _tensor_regs.cend();
  }

getBackPropITensor()

backend::ITensor * onert::compiler::train::TensorRegistries::getBackPropITensor ( ir::OperandIndex  index ) const

inline

Definition at line 86 of file TensorRegistries.h.

  {
    for (const auto &tensor_reg : _tensor_regs)
    {
      auto tensor = tensor_reg->getBackPropITensor(index);
      if (tensor)
        return tensor;
    }
    return nullptr;
  }

getBuiltinTensorRegistry()

std::shared_ptr< backend::builtin::train::TensorRegistry > onert::compiler::train::TensorRegistries::getBuiltinTensorRegistry ( ) const

inline

Definition at line 70 of file TensorRegistries.h.

  {
    return _builtin_tensor_reg;
  }

getITensor()

backend::ITensor * onert::compiler::train::TensorRegistries::getITensor ( ir::OperandIndex  index ) const

inline

Definition at line 75 of file TensorRegistries.h.

  {
    for (const auto &tensor_reg : _tensor_regs)
    {
      auto tensor = tensor_reg->getITensor(index);
      if (tensor)
        return tensor;
    }
    return nullptr;
  }

Referenced by onert::exec::train::TrainableExecutor::getLoss(), and onert::exec::train::TrainableExecutor::TrainableExecutor().

iterateTrainableTensors()

void onert::compiler::train::TensorRegistries::iterateTrainableTensors ( const std::function< void(const ir::OperandIndex &, const backend::train::ITrainableTensor *)> &  fn ) const

inline

Definition at line 97 of file TensorRegistries.h.

  {
    for (const auto &tensor_reg : _tensor_regs)
      tensor_reg->iterateTrainableTensors(fn);
  }

Referenced by onert::exec::train::TrainableExecutor::iterateTrainableTensors().

---

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

• runtime/onert/core/src/compiler/train/TensorRegistries.h