#include <BackendContext.h>

Collaboration diagram for onert::backend::cpu::BackendContext:

[legend]

Public Member Functions
	BackendContext (const Backend *backend, ContextData &&data, std::shared_ptr< ITensorRegistry > tensor_registry=nullptr, std::shared_ptr< TensorBuilder > tensor_builder=nullptr, std::shared_ptr< KernelGenerator > kernel_gen=nullptr)

ITensorRegistry *	genTensors () override

FunctionMap	genKernels () override

std::shared_ptr< ExternalContext >	external_context ()

Public Member Functions inherited from onert::backend::BackendContext
	BackendContext (const Backend *backend, ContextData &&data, std::shared_ptr< ITensorRegistry > tensor_registry=nullptr)

virtual	~BackendContext ()=default

const Backend *	backend () const

const ir::Graph *	graph () const

const util::Set< ir::OperandIndex > &	external_operands () const

const ContextData &	data () const

Data Fields
std::shared_ptr< TensorBuilder >	tensor_builder

std::shared_ptr< KernelGenerator >	kernel_gen

Data Fields inherited from onert::backend::BackendContext
std::shared_ptr< ITensorRegistry >	tensor_registry

Additional Inherited Members
Protected Attributes inherited from onert::backend::BackendContext
const Backend *	_backend {nullptr}

ContextData	_data

Detailed Description

Definition at line 32 of file BackendContext.h.

Constructor & Destructor Documentation

◆ BackendContext()

onert::backend::cpu::BackendContext::BackendContext	(	const Backend *	backend,
		ContextData &&	data,
		std::shared_ptr< ITensorRegistry >	tensor_registry = `nullptr`,
		std::shared_ptr< TensorBuilder >	tensor_builder = `nullptr`,
		std::shared_ptr< KernelGenerator >	kernel_gen = `nullptr`
	)

inline

Definition at line 35 of file BackendContext.h.

    : onert::backend::BackendContext(backend, std::move(data), tensor_registry),
      tensor_builder{tensor_builder}, kernel_gen{kernel_gen}, _external_context(new ExternalContext)
  {
  }

Member Function Documentation

◆ external_context()

std::shared_ptr< ExternalContext > onert::backend::cpu::BackendContext::external_context ( )

inline

Definition at line 47 of file BackendContext.h.

47{ return _external_context; }

◆ genKernels()

FunctionMap onert::backend::cpu::BackendContext::genKernels ( )

overridevirtual

Implements onert::backend::BackendContext.

Definition at line 43 of file BackendContext.cc.

{
  FunctionMap ret;
 
  basic::initConsts(graph()->operands(), external_operands(), tensor_registry.get(),
                    tensor_builder->getSharedMemoryOperandIndexes());
 
  // TODO: Change type of tensor_registry field to TensorRegistry
  auto tensor_registry_concreted =
    nnfw::misc::polymorphic_downcast<basic::TensorRegistry *>(tensor_registry.get());
  basic::initSharedMemoryConsts(graph()->operands(), external_operands(), tensor_registry_concreted,
                                tensor_builder->getSharedMemoryOperandIndexes());
 
  for (auto &&op_ind : _data.op_order)
  {
    auto fn_seq = kernel_gen->generate(op_ind);
    ret.emplace(op_ind, std::move(fn_seq));
  }
 
  // NOTE For memory optimization, we want to free some operand data
  const_cast<ir::Graph &>(*_data.graph)
    .operands()
    .iterate([&](const ir::OperandIndex &, ir::Operand &obj) { obj.releaseData(); });
 
  for (auto &&it : ret)
  {
    auto &fn_seq = it.second;
    fn_seq->iterate([&](exec::IFunction &ifunc) { ifunc.prepare(); });
  }
 
  return ret;
}

References onert::backend::BackendContext::_data, onert::backend::BackendContext::external_operands(), onert::backend::ContextData::graph, onert::backend::BackendContext::graph(), onert::backend::basic::initConsts(), onert::backend::basic::initSharedMemoryConsts(), kernel_gen, onert::backend::ContextData::op_order, onert::exec::IFunction::prepare(), tensor_builder, and onert::backend::BackendContext::tensor_registry.

◆ genTensors()

ITensorRegistry * onert::backend::cpu::BackendContext::genTensors ( )

overridevirtual

Implements onert::backend::BackendContext.

Definition at line 37 of file BackendContext.cc.

{
  return basic::genTensors(tensor_builder, *graph(), external_operands(), tensor_registry,
                           data().op_order, tensor_builder->getSharedMemoryOperandIndexes());
}

References onert::backend::BackendContext::data(), onert::backend::BackendContext::external_operands(), onert::backend::basic::genTensors(), onert::backend::BackendContext::graph(), tensor_builder, and onert::backend::BackendContext::tensor_registry.

Field Documentation

◆ kernel_gen

std::shared_ptr<KernelGenerator> onert::backend::cpu::BackendContext::kernel_gen

Definition at line 52 of file BackendContext.h.

Referenced by genKernels().

◆ tensor_builder

std::shared_ptr<TensorBuilder> onert::backend::cpu::BackendContext::tensor_builder

Definition at line 51 of file BackendContext.h.

Referenced by genKernels(), and genTensors().

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

runtime/onert/backend/cpu/BackendContext.h
runtime/onert/backend/cpu/BackendContext.cc

Public Member Functions

Data Fields

Additional Inherited Members