From 3ada2b7a29e1ab2058ab7dc701cacff548d2aae9 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 23 Aug 2018 15:54:36 +0100 Subject: COMPMID-1534: Fix LSTM/RNN Layers for NEON and FP16 Switches default activation layer to the respective datasets to RELU from LOGISTIC Change-Id: I09f1ad09922ccdd6e1dc33c28a594f7ffbfe40f4 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145436 Reviewed-by: Anthony Barbier Tested-by: Jenkins --- arm_compute/runtime/NEON/functions/NERNNLayer.h | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'arm_compute/runtime/NEON/functions/NERNNLayer.h') diff --git a/arm_compute/runtime/NEON/functions/NERNNLayer.h b/arm_compute/runtime/NEON/functions/NERNNLayer.h index f1398eb3cc..bdba42d6ba 100644 --- a/arm_compute/runtime/NEON/functions/NERNNLayer.h +++ b/arm_compute/runtime/NEON/functions/NERNNLayer.h @@ -26,6 +26,7 @@ #include "arm_compute/core/NEON/kernels/NEActivationLayerKernel.h" #include "arm_compute/core/NEON/kernels/NEArithmeticAdditionKernel.h" +#include "arm_compute/core/NEON/kernels/NECopyKernel.h" #include "arm_compute/runtime/NEON/INESimpleFunction.h" #include "arm_compute/core/Types.h" @@ -79,6 +80,7 @@ public: // Inherited methods overridden: void run() override; + void prepare() override; private: MemoryGroup _memory_group; @@ -86,11 +88,11 @@ private: NEArithmeticAdditionKernel _add_kernel; NEActivationLayerKernel _activation_kernel; NEFullyConnectedLayer _fully_connected_kernel; + NECopyKernel _copy_kernel; Tensor _fully_connected_out; Tensor _gemm_output; Tensor _add_output; - ITensor *_hidden_state; - ITensor *_output; + bool _is_prepared; }; } // namespace arm_compute #endif /* __ARM_COMPUTE_NERNNLAYER_H__ */ -- cgit v1.2.1