From db2ef067e585e03d4d3f8cd4401babca44671d4a Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Principal=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Wed, 18 Apr 2018 12:52:28 +0900 Subject: [PATCH] Register concat to nnfw kernel table (#767) This will add concatenationFloat32 on ACL to kernel lookup table Signed-off-by: SaeHie Park --- src/runtime/ref/nn/common/NNFWKernels.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/runtime/ref/nn/common/NNFWKernels.cpp b/src/runtime/ref/nn/common/NNFWKernels.cpp index 5fa24ef..61e8c56 100644 --- a/src/runtime/ref/nn/common/NNFWKernels.cpp +++ b/src/runtime/ref/nn/common/NNFWKernels.cpp @@ -9,6 +9,7 @@ #include "kernel/acl/Pooling.h" #include "kernel/acl/Softmax.h" #include "kernel/acl/FullyConnected.h" +#include "kernel/acl/Concatenation.h" #include "kernel/acl/nnfw_kernel_acl.h" #endif // USE_NNFW_ACL_KERNELS @@ -33,6 +34,7 @@ void init_nnfw_kernels() nnfw_kernels_maxPoolFloat32["acl"] = nnfw::kernel::acl::maxPoolFloat32; nnfw_kernels_softmaxFloat32["acl"] = nnfw::kernel::acl::softmaxFloat32; nnfw_kernels_fullyConnectedFloat32["acl"] = nnfw::kernel::acl::fullyConnectedFloat32; + nnfw_kernels_concatenationFloat32["acl"] = nnfw::kernel::acl::concatenationFloat32; #endif // USE_NNFW_ACL_KERNELS return; } -- 2.7.4