diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-08-23 15:54:36 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | 3ada2b7a29e1ab2058ab7dc701cacff548d2aae9 (patch) | |
| tree | 734a55577ca01011de6d7f1587afbf65b0ab1f38 /arm_compute/runtime/NEON/functions/NERNNLayer.h | |
| parent | f2cde9b29deee6423ea6fe9a1a9afc9ef61d2663 (diff) | |
| download | ComputeLibrary-3ada2b7a29e1ab2058ab7dc701cacff548d2aae9.tar.gz | |
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 <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NERNNLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NERNNLayer.h | 6 |
1 files changed, 4 insertions, 2 deletions
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__ */ |