onert::backend::train::KernelGeneratorBase Class Reference

#include <KernelGeneratorBase.h>

Collaboration diagram for onert::backend::train::KernelGeneratorBase:

Public Member Functions
virtual	~KernelGeneratorBase ()=default
	KernelGeneratorBase (const ir::train::TrainableGraph &tgraph)
virtual std::unique_ptr< exec::train::TrainableFnSequence >	generate (ir::OperationIndex ind)=0
Public Member Functions inherited from onert::ir::train::TrainableOperationVisitor
virtual	~TrainableOperationVisitor ()=default

Protected Attributes
const ir::train::TrainableGraph &	_tgraph
std::unique_ptr< exec::train::ITrainableFunction >	_return_fn

Detailed Description

Definition at line 34 of file KernelGeneratorBase.h.

Constructor & Destructor Documentation

~KernelGeneratorBase()

virtual onert::backend::train::KernelGeneratorBase::~KernelGeneratorBase ( )

virtualdefault

KernelGeneratorBase()

onert::backend::train::KernelGeneratorBase::KernelGeneratorBase ( const ir::train::TrainableGraph &  tgraph )

inline

Definition at line 38 of file KernelGeneratorBase.h.

: _tgraph{tgraph} {}

Member Function Documentation

generate()

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

pure virtual

Implemented in onert::backend::builtin::train::KernelGenerator, and onert::backend::train::KernelGenerator.

Field Documentation

_return_fn

std::unique_ptr<exec::train::ITrainableFunction> onert::backend::train::KernelGeneratorBase::_return_fn

protected

Definition at line 53 of file KernelGeneratorBase.h.

Referenced by onert::backend::builtin::train::KernelGenerator::generate(), onert::backend::train::KernelGenerator::generate(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), and onert::backend::train::KernelGenerator::visit().

_tgraph

const ir::train::TrainableGraph& onert::backend::train::KernelGeneratorBase::_tgraph

protected

Definition at line 52 of file KernelGeneratorBase.h.

Referenced by onert::backend::builtin::train::KernelGenerator::generate(), onert::backend::train::KernelGenerator::generate(), onert::backend::train::KernelGenerator::visit(), onert::backend::train::KernelGenerator::visit(), and onert::backend::train::KernelGenerator::visit().

---

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

• runtime/onert/core/include/backend/train/KernelGeneratorBase.h