From 6844b8800128d27b469ae4905756deea90dd791c Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Fri, 8 Feb 2019 16:56:00 +0000 Subject: COMPMID-1710 Fix NEON LSTM invalid allocation Change-Id: I5f9d435df63f351062c7091aa4d1d3910a01743e Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/653 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Isabella Gottardi --- src/runtime/NEON/functions/NELSTMLayer.cpp | 1 - 1 file changed, 1 deletion(-) diff --git a/src/runtime/NEON/functions/NELSTMLayer.cpp b/src/runtime/NEON/functions/NELSTMLayer.cpp index 73afd054fd..9e7a713ada 100644 --- a/src/runtime/NEON/functions/NELSTMLayer.cpp +++ b/src/runtime/NEON/functions/NELSTMLayer.cpp @@ -130,7 +130,6 @@ void NELSTMLayer::configure(const ITensor *input, _forget_gate_out3.allocator()->allocate(); } _activation_forget_gate.configure(forget_gate_out, nullptr, ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::LOGISTIC)); - forget_gate_out->allocator()->allocate(); // Configure block that calculates the input gate // input_gate = Activation(input * input_to_input_weights + output_state * recurrent_to_input_weights + PixelWiseMul(cell_state, cell_to_input_weights) + input_gate_bias), without CIFG -- cgit v1.2.1