diff options
Diffstat (limited to 'runtime/onert/backend/acl_neon/KernelGenerator.h')
| -rw-r--r-- | runtime/onert/backend/acl_neon/KernelGenerator.h | 25 |
1 files changed, 13 insertions, 12 deletions
diff --git a/runtime/onert/backend/acl_neon/KernelGenerator.h b/runtime/onert/backend/acl_neon/KernelGenerator.h index 4d269cde5..2a4b307b8 100644 --- a/runtime/onert/backend/acl_neon/KernelGenerator.h +++ b/runtime/onert/backend/acl_neon/KernelGenerator.h @@ -17,7 +17,7 @@ #ifndef __ONERT_BACKEND_ACL_NEON_KERNEL_GENERATOR_H__ #define __ONERT_BACKEND_ACL_NEON_KERNEL_GENERATOR_H__ -#include <backend/IKernelGenerator.h> +#include <backend/cpu_common/KernelGeneratorBase.h> #include "ir/Operands.h" #include "TensorBuilder.h" @@ -31,7 +31,7 @@ namespace backend namespace acl_neon { -class KernelGenerator : public IKernelGenerator +class KernelGenerator : public cpu_common::KernelGeneratorBase { public: KernelGenerator(const ir::Operands &operands_ctx, const ir::Operations &operations_ctx, @@ -39,17 +39,20 @@ public: const std::shared_ptr<acl_common::AclTensorRegistry<TensorManager>> &_tensor_reg); void visit(const ir::OpSequence &) override; - void visit(const ir::operation::ArgMax &) override; + + void visit(const ir::operation::ArgMinMax &) override; void visit(const ir::operation::BatchToSpaceND &) override; void visit(const ir::operation::BinaryArithmetic &) override; + void visit(const ir::operation::Comparison &) override; + void visit(const ir::operation::Concat &) override; void visit(const ir::operation::Conv2D &) override; void visit(const ir::operation::DepthToSpace &) override; void visit(const ir::operation::DepthwiseConv2D &) override; - void visit(const ir::operation::Concat &) override; void visit(const ir::operation::ElementwiseActivation &) override; void visit(const ir::operation::ElementwiseBinary &) override; void visit(const ir::operation::ElementwiseUnary &) override; void visit(const ir::operation::EmbeddingLookup &) override; + void visit(const ir::operation::ExpandDims &) override; void visit(const ir::operation::FullyConnected &) override; void visit(const ir::operation::Gather &) override; void visit(const ir::operation::HashtableLookup &) override; @@ -57,36 +60,34 @@ public: void visit(const ir::operation::L2Normalization &) override; void visit(const ir::operation::LocalResponseNormalization &) override; void visit(const ir::operation::LSTM &) override; + void visit(const ir::operation::OneHot &) override; void visit(const ir::operation::Pack &) override; void visit(const ir::operation::Pad &) override; - void visit(const ir::operation::Pool2D &) override; void visit(const ir::operation::Permute &) override; + void visit(const ir::operation::Pool2D &) override; void visit(const ir::operation::PReLU &) override; void visit(const ir::operation::Reduce &) override; void visit(const ir::operation::Reshape &) override; void visit(const ir::operation::ResizeBilinear &) override; void visit(const ir::operation::RNN &) override; - void visit(const ir::operation::Squeeze &) override; + void visit(const ir::operation::Slice &) override; void visit(const ir::operation::Softmax &) override; void visit(const ir::operation::SpaceToBatchND &) override; void visit(const ir::operation::SpaceToDepth &) override; void visit(const ir::operation::Split &) override; void visit(const ir::operation::SquaredDifference &) override; - void visit(const ir::operation::Slice &) override; + void visit(const ir::operation::Squeeze &) override; void visit(const ir::operation::StridedSlice &) override; - void visit(const ir::operation::TransposeConv &) override; void visit(const ir::operation::Transpose &) override; + void visit(const ir::operation::TransposeConv &) override; void visit(const ir::operation::Unpack &) override; - void visit(const ir::operation::ExpandDims &) override; - void visit(const ir::operation::Comparison &) override; - void visit(const ir::operation::OneHot &) override; private: const ir::Operands &_ctx; const ir::Operations &_operations_ctx; std::shared_ptr<TensorBuilder> _tensor_builder; std::shared_ptr<acl_common::AclTensorRegistry<TensorManager>> _tensor_reg; - ir::Layout _current_op_seq_layout; + ir::Layout _current_layout; }; } // namespace acl_neon |