Enable acl/neon backend (#776)
author박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Wed, 18 Apr 2018 06:51:21 +0000 (15:51 +0900)
committer김정현/동작제어Lab(SR)/Senior Engineer/삼성전자 <jh0822.kim@samsung.com>
Wed, 18 Apr 2018 06:51:21 +0000 (15:51 +0900)
This will add acl/neon kernel implementations to kernel lookup tables

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
src/runtime/ref/nn/common/NNFWKernels.cpp

index 61e8c56..636ba92 100644 (file)
@@ -35,6 +35,13 @@ void init_nnfw_kernels()
   nnfw_kernels_softmaxFloat32["acl"] = nnfw::kernel::acl::softmaxFloat32;
   nnfw_kernels_fullyConnectedFloat32["acl"] = nnfw::kernel::acl::fullyConnectedFloat32;
   nnfw_kernels_concatenationFloat32["acl"] = nnfw::kernel::acl::concatenationFloat32;
+
+  nnfw_kernels_convFloat32["neon"] = nnfw::kernel::acl::neon::convFloat32;
+  nnfw_kernels_averagePoolFloat32["neon"] = nnfw::kernel::acl::neon::averagePoolFloat32;
+  nnfw_kernels_maxPoolFloat32["neon"] = nnfw::kernel::acl::neon::maxPoolFloat32;
+  nnfw_kernels_softmaxFloat32["neon"] = nnfw::kernel::acl::neon::softmaxFloat32;
+  nnfw_kernels_fullyConnectedFloat32["neon"] = nnfw::kernel::acl::neon::fullyConnectedFloat32;
+  nnfw_kernels_concatenationFloat32["neon"] = nnfw::kernel::acl::neon::concatenationFloat32;
 #endif // USE_NNFW_ACL_KERNELS
   return;
 }