ONE - On-device Neural Engine
|
#include <KernelGenerator.h>
Public Member Functions | |
KernelGenerator (const ir::Graph &graph, const std::shared_ptr< TensorBuilder > &tensor_builder, const std::shared_ptr< acl_common::AclTensorRegistry< TensorManager > > &_tensor_reg) | |
std::unique_ptr< exec::FunctionSequence > | generate (ir::OperationIndex ind) override |
Public Member Functions inherited from onert::backend::basic::KernelGeneratorBase | |
virtual | ~KernelGeneratorBase ()=default |
KernelGeneratorBase (const ir::Graph &graph) | |
Public Member Functions inherited from onert::ir::OperationVisitor | |
virtual | ~OperationVisitor ()=default |
Additional Inherited Members | |
Protected Member Functions inherited from onert::backend::basic::KernelGeneratorBase | |
std::unique_ptr< exec::IFunction > | releaseFunction () |
Protected Attributes inherited from onert::backend::basic::KernelGeneratorBase | |
const ir::Graph & | _graph |
std::unique_ptr< exec::IFunction > | _return_fn |
Definition at line 33 of file KernelGenerator.h.
onert::backend::acl_cl::KernelGenerator::KernelGenerator | ( | const ir::Graph & | graph, |
const std::shared_ptr< TensorBuilder > & | tensor_builder, | ||
const std::shared_ptr< acl_common::AclTensorRegistry< TensorManager > > & | _tensor_reg | ||
) |
Definition at line 46 of file KernelGenerator.cc.
|
overridevirtual |
Implements onert::backend::basic::KernelGeneratorBase.
Definition at line 55 of file KernelGenerator.cc.
References onert::backend::basic::KernelGeneratorBase::_graph, onert::util::ObjectManager< Index, Object >::at(), onert::ir::Graph::operations(), and onert::backend::basic::KernelGeneratorBase::releaseFunction().