ONE - On-device Neural Engine
|
#include <BackendContext.h>
Public Member Functions | |
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 |
virtual ITensorRegistry * | genTensors ()=0 |
virtual FunctionMap | genKernels ()=0 |
Data Fields | |
std::shared_ptr< ITensorRegistry > | tensor_registry |
Protected Attributes | |
const Backend * | _backend {nullptr} |
ContextData | _data |
Definition at line 51 of file BackendContext.h.
|
inline |
Definition at line 54 of file BackendContext.h.
|
virtualdefault |
|
inline |
Definition at line 62 of file BackendContext.h.
References _backend.
Referenced by CfgRunner.CfgRunner::run(), and CfgRunner.CfgRunner::set_backend().
|
inline |
Definition at line 65 of file BackendContext.h.
References _data.
Referenced by onert::backend::cpu::BackendContext::genTensors().
|
inline |
Definition at line 64 of file BackendContext.h.
References _data, and onert::backend::ContextData::external_operands.
Referenced by onert::backend::cpu::BackendContext::genKernels(), onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::genTensors(), onert::backend::cpu::BackendContext::genTensors(), and onert::backend::basic::initConsts().
|
pure virtual |
Implemented in onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >, onert::backend::cpu::BackendContext, onert::backend::ruy::BackendContext, onert::backend::train::DummyBackendContext, onert::backend::trix::BackendContext, onert::backend::xnnpack::BackendContext, and onert::backend::builtin::BackendContext.
|
pure virtual |
Implemented in onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >, onert::backend::cpu::BackendContext, onert::backend::ruy::BackendContext, onert::backend::train::DummyBackendContext, onert::backend::trix::BackendContext, onert::backend::xnnpack::BackendContext, and onert::backend::builtin::BackendContext.
|
inline |
Definition at line 63 of file BackendContext.h.
References _data, and onert::backend::ContextData::graph.
Referenced by TopologicalSortHelper.TopologicalSortHelper::add_edge(), onert::backend::cpu::BackendContext::genKernels(), onert::backend::builtin::BackendContext::genKernels(), onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::genTensors(), onert::backend::cpu::BackendContext::genTensors(), onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >::initConsts(), onert::backend::basic::initConsts(), onert::backend::acl_cl::Optimizer::optimize(), onert::backend::acl_neon::Optimizer::optimize(), onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >::planTensors(), and TopologicalSortHelper.TopologicalSortHelper::sort_util().
|
protected |
|
protected |
Definition at line 72 of file BackendContext.h.
Referenced by data(), external_operands(), onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >::genKernels(), onert::backend::cpu::BackendContext::genKernels(), onert::backend::ruy::BackendContext::genKernels(), onert::backend::trix::BackendContext::genKernels(), onert::backend::xnnpack::BackendContext::genKernels(), onert::backend::builtin::BackendContext::genKernels(), graph(), onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >::initConsts(), and onert::backend::cl_common::BackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator >::planTensors().
std::shared_ptr<ITensorRegistry> onert::backend::BackendContext::tensor_registry |
Definition at line 75 of file BackendContext.h.
Referenced by onert::backend::cpu::BackendContext::genKernels(), onert::backend::acl_common::AclBackendContext< T_TensorBuilder, T_ConstantInitializer, T_KernelGenerator, T_Optimizer >::genTensors(), onert::backend::cpu::BackendContext::genTensors(), and onert::backend::basic::initConsts().