onert::backend::train::Backend Class Reference

#include <Backend.h>

Collaboration diagram for onert::backend::train::Backend:

Public Member Functions
	Backend ()
std::shared_ptr< IConfig >	config () const override
std::unique_ptr< onert::backend::BackendContext >	newContext (ContextData &&data) const override
std::unique_ptr< backend::train::TrainableBackendContext >	newContext (backend::train::TrainableContextData &&tdata) const override
Public Member Functions inherited from onert::backend::Backend
virtual	~Backend ()=default
Public Member Functions inherited from onert::backend::train::ITrainableBackend
virtual	~ITrainableBackend ()=default

Detailed Description

Definition at line 35 of file Backend.h.

Constructor & Destructor Documentation

Backend()

onert::backend::train::Backend::Backend ( )

inline

Definition at line 38 of file Backend.h.

: _config{std::make_shared<Config>()} {}

Member Function Documentation

config()

std::shared_ptr< IConfig > onert::backend::train::Backend::config ( ) const

inlineoverridevirtual

Implements onert::backend::Backend.

Definition at line 40 of file Backend.h.

{ return _config; }

newContext() [1/2]

std::unique_ptr< backend::train::TrainableBackendContext > onert::backend::train::Backend::newContext ( backend::train::TrainableContextData &&  tdata ) const

inlineoverridevirtual

Implements onert::backend::train::ITrainableBackend.

Definition at line 48 of file Backend.h.

  {
    const auto &tgraph = *tdata.tgraph;
    auto optimizer = createOptimizer(tdata.optim_info);
    auto tr = std::make_shared<TensorRegistry>();
    auto tb = std::make_shared<TensorBuilder>(tr, optimizer.get());
    auto tdata_ptr = std::make_unique<backend::train::TrainableContextData>(std::move(tdata));
    auto context = std::make_unique<train::BackendContext>(this, std::move(tdata_ptr), tr, tb,
                                                           std::move(optimizer));
 
    context->kernel_gen = std::make_shared<train::KernelGenerator>(
      tgraph, tr, context->external_context(), context->optimizer());
    return context;
  }

References onert::backend::train::createOptimizer().

newContext() [2/2]

std::unique_ptr< onert::backend::BackendContext > onert::backend::train::Backend::newContext ( ContextData &&  data ) const

inlineoverridevirtual

Implements onert::backend::Backend.

Definition at line 42 of file Backend.h.

  {
    return std::make_unique<DummyBackendContext>(this, std::move(data));
  }

---

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

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