onert::backend::builtin::Backend Class Reference

#include <Backend.h>

Collaboration diagram for onert::backend::builtin::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 41 of file Backend.h.

Constructor & Destructor Documentation

Backend()

onert::backend::builtin::Backend::Backend ( )

inline

Definition at line 44 of file Backend.h.

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

Member Function Documentation

config()

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

inlineoverridevirtual

Implements onert::backend::Backend.

Definition at line 46 of file Backend.h.

{ return _config; }

newContext() [1/2]

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

inlineoverridevirtual

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

Definition at line 78 of file Backend.h.

  {
    const auto &tgraph = *tdata.tgraph;
    auto tr = std::make_shared<train::TensorRegistry>();
    // TODO Create TensorBuilder if necessary
    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);
 
    context->kernel_gen =
      std::make_shared<train::KernelGenerator>(tgraph, tr, context->external_context());
    return context;
  }

newContext() [2/2]

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

inlineoverridevirtual

Implements onert::backend::Backend.

Definition at line 48 of file Backend.h.

  {
    auto context = std::make_unique<BackendContext>(this, std::move(data));
    // ControlFlow backend may not build tensors for itself because the backend's operation uses
    // tensors of other baceknd instead
    // But the backend builds tensors in case of that the controlflow operation may have constant
    // input or that consecutive controflow operations exist. We have to make them not to be built
    // later
    // 1. Constant input
    //   These tensors cannot be dynamic tensor, so let's do it as follows:
    //   - always skip copying
    //   - if it is operation's input in child subgraph: register "use" as constant input of the
    //   operations in child subgraph
    //   - if it is child subgraph's output: register "use" as constant input of the operations
    //   using it
    // 2. Consecutive controflow operation's intermediate tensor
    //   These tensors can be dynamic tensor and this is complicated to support without copying. But
    //   there is no such case until now, let's support it later
    // TODO Remove TensorBuilder and ConstantInitializer
    // TODO Support Consecutive controflow operation's intermediate tensor
    auto tr = std::make_shared<TensorRegistry>();
    auto tb = std::make_shared<TensorBuilder>(tr);
    context->tensor_registry = tr;
    context->tensor_builder = tb;
    context->kernel_gen = std::make_shared<KernelGenerator>(
      *context->graph(), tb->dynamicTensorManager(), tr, context->external_context());
    return context;
  }

---

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

• runtime/onert/core/src/backend/builtin/Backend.h