onert::backend::builtin::train::KernelGenerator Class Reference

#include <KernelGenerator.h>

Collaboration diagram for onert::backend::builtin::train::KernelGenerator:

Public Member Functions
	KernelGenerator (const ir::train::TrainableGraph &tgraph, const std::shared_ptr< TensorRegistry > &tensor_reg, const std::shared_ptr< ExternalContext > &external_context)
std::unique_ptr< exec::train::TrainableFnSequence >	generate (ir::OperationIndex ind) override
void	setTensorRegistries (const compiler::train::TensorRegistries &tensor_registries)
void	setWholeGraphOutputs (const ir::OperandIndexSequence &outputs)
Public Member Functions inherited from onert::backend::train::KernelGeneratorBase
virtual	~KernelGeneratorBase ()=default
	KernelGeneratorBase (const ir::train::TrainableGraph &tgraph)
Public Member Functions inherited from onert::ir::train::TrainableOperationVisitor
virtual	~TrainableOperationVisitor ()=default

Additional Inherited Members
Protected Attributes inherited from onert::backend::train::KernelGeneratorBase
const ir::train::TrainableGraph &	_tgraph
std::unique_ptr< exec::train::ITrainableFunction >	_return_fn

Detailed Description

Definition at line 31 of file KernelGenerator.h.

Constructor & Destructor Documentation

KernelGenerator()

onert::backend::builtin::train::KernelGenerator::KernelGenerator	(	const ir::train::TrainableGraph &	tgraph,
		const std::shared_ptr< TensorRegistry > &	tensor_reg,
		const std::shared_ptr< ExternalContext > &	external_context
	)

Definition at line 24 of file KernelGenerator.cc.

  : KernelGeneratorBase{tgraph}, _tensor_reg{tensor_reg}, _external_context(external_context)
{
}

Member Function Documentation

generate()

std::unique_ptr< exec::train::TrainableFnSequence > onert::backend::builtin::train::KernelGenerator::generate ( ir::OperationIndex  ind )

overridevirtual

Implements onert::backend::train::KernelGeneratorBase.

Definition at line 31 of file KernelGenerator.cc.

{
  auto ret = std::make_unique<exec::train::TrainableFnSequence>();
  const auto &op = _tgraph.operation(ind);
  op.accept(*this);
  // _return_fn must have been generated
  if (_return_fn == nullptr)
  {
    throw std::runtime_error(op.name() + " op does not supported trainable kernel yet");
  }
 
  ret->_functions.emplace_back(std::move(_return_fn));
 
  return ret;
}

References onert::backend::train::KernelGeneratorBase::_return_fn, onert::backend::train::KernelGeneratorBase::_tgraph, onert::ir::train::ITrainableOperation::accept(), and onert::ir::train::TrainableGraph::operation().

setTensorRegistries()

void onert::backend::builtin::train::KernelGenerator::setTensorRegistries ( const compiler::train::TensorRegistries &  tensor_registries )

inline

Definition at line 40 of file KernelGenerator.h.

  {
    _tensor_registries = tensor_registries;
  }

setWholeGraphOutputs()

void onert::backend::builtin::train::KernelGenerator::setWholeGraphOutputs ( const ir::OperandIndexSequence &  outputs )

inline

Definition at line 45 of file KernelGenerator.h.

  {
    _whole_graph_outputs = outputs;
  }

---

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

• runtime/onert/core/src/backend/builtin/train/KernelGenerator.h
• runtime/onert/core/src/backend/builtin/train/KernelGenerator.cc