onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer > Class Template Reference

#include <AclBackendContext.h>

Collaboration diagram for onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >:

Public Member Functions
	AclBackendContext (const Backend *backend, ContextData &&data, std::shared_ptr< ITensorRegistry > tensor_registry=nullptr, std::shared_ptr< T_TensorBuilder > tensor_builder=nullptr, std::shared_ptr< T_ConstantInitializer > constant_initializer=nullptr, std::shared_ptr< T_KernelGenerator > kernel_gen=nullptr)
ITensorRegistry *	genTensors () override
Public Member Functions inherited from onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >
	BackendContext (const Backend *backend, ContextData &&data, std::shared_ptr< ITensorRegistry > tensor_registry=nullptr, std::shared_ptr< T_TensorBuilder > tensor_builder=nullptr, std::shared_ptr< T_ConstantInitializer > constant_initializer=nullptr, std::shared_ptr< T_KernelGenerator > kernel_gen=nullptr)
FunctionMap	genKernels () override
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< T_Optimizer >	optimizer
Data Fields inherited from onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >
std::shared_ptr< T_TensorBuilder >	tensor_builder
std::shared_ptr< T_ConstantInitializer >	constant_initializer
std::shared_ptr< T_KernelGenerator >	kernel_gen
Data Fields inherited from onert::backend::BackendContext
std::shared_ptr< ITensorRegistry >	tensor_registry

Protected Member Functions
void	registerTensorInfo (const ir::OperandIndex &ind, const ir::OperandInfo &info) override
Protected Member Functions inherited from onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >
void	initConsts ()
void	planTensors ()

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

Detailed Description

template<typename T_TensorBuilder, typename T_ConstantInitializer, typename T_KernelGenerator, typename T_Optimizer>
class onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >

Definition at line 34 of file AclBackendContext.h.

Constructor & Destructor Documentation

AclBackendContext()

template<typename T_TensorBuilder , typename T_ConstantInitializer , typename T_KernelGenerator , typename T_Optimizer >

onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::AclBackendContext ( const Backend *  backend, ContextData &&  data, std::shared_ptr< ITensorRegistry >  tensor_registry = nullptr, std::shared_ptr< T_TensorBuilder >  tensor_builder = nullptr, std::shared_ptr< T_ConstantInitializer >  constant_initializer = nullptr, std::shared_ptr< T_KernelGenerator >  kernel_gen = nullptr  )

inline

Definition at line 39 of file AclBackendContext.h.

    : onert::backend::cl_common::BackendContext<T_TensorBuilder, T_ConstantInitializer,
                                                T_KernelGenerator>(
        backend, std::move(data), tensor_registry, tensor_builder, constant_initializer, kernel_gen)
  {
    // DO NOTHING
  }

Member Function Documentation

genTensors()

template<typename T_TensorBuilder , typename T_ConstantInitializer , typename T_KernelGenerator , typename T_Optimizer >

ITensorRegistry * onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::genTensors ( )

inlineoverridevirtual

Implements onert::backend::BackendContext.

Definition at line 51 of file AclBackendContext.h.

  {
    optimizer->optimize();
 
    this->graph()->operands().iterate([&](const ir::OperandIndex &ind, const ir::Operand &obj) {
      if (this->external_operands().contains(ind))
        return;
 
      ir::OperandInfo backend_info{obj.shape(), obj.typeInfo(), obj.info().memAllocType(),
                                   obj.isConstant()};
      this->tensor_builder->registerTensorInfo(ind, backend_info);
    });
 
    // TODO Get compiler options from compiler, and use it rather than getting it from Env
    if (util::getConfigString(util::config::EXECUTOR) == "Linear")
    {
      this->planTensors();
    }
    else
    {
      // For the executors that does not have fixed linear execution order:
      // To make tensors never be deallocated, this is a workaround to use static memory planner
      this->graph()->operands().iterate([&](const ir::OperandIndex &ind, const ir::Operand &) {
        if (this->tensor_builder->isRegistered(ind))
          this->tensor_builder->notifyFirstUse(ind);
      });
    }
 
    this->tensor_builder->prepare();
 
    return this->tensor_registry.get();
  }

References onert::backend::BackendContext::external_operands(), onert::util::getConfigString(), onert::backend::BackendContext::graph(), onert::util::ObjectManager< Index, Object >::iterate(), onert::ir::Graph::operands(), onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::optimizer, onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >::planTensors(), onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >::tensor_builder, and onert::backend::BackendContext::tensor_registry.

registerTensorInfo()

template<typename T_TensorBuilder , typename T_ConstantInitializer , typename T_KernelGenerator , typename T_Optimizer >

void onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::registerTensorInfo ( const ir::OperandIndex &  ind, const ir::OperandInfo &  info  )

inlineoverrideprotectedvirtual

Implements onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >.

Definition at line 85 of file AclBackendContext.h.

  {
    this->tensor_builder->registerTensorInfo(ind, info);
  }

References info, and onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >::tensor_builder.

Field Documentation

optimizer

template<typename T_TensorBuilder , typename T_ConstantInitializer , typename T_KernelGenerator , typename T_Optimizer >

std::shared_ptr<T_Optimizer> onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::optimizer

Definition at line 92 of file AclBackendContext.h.

Referenced by onert.experimental.train.session.TrainSession::__init__(), onert.experimental.train.session.TrainSession::_print_training_parameters(), onert.experimental.train.session.TrainSession::eval_step(), onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::genTensors(), onert.experimental.train.session.TrainSession::train(), and onert.experimental.train.session.TrainSession::train_step().

---

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

• runtime/onert/backend/acl_common/AclBackendContext.h