From 2761c2f0b60175469e959982a25ff0abdca6c9ce Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Fri, 22 Mar 2019 13:06:08 +0000 Subject: COMPMID-2075 Segfault in NELSTM Change-Id: Ice7d7cc878614b5e317c0d006c4429be27f5dc29 Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/c/888 Reviewed-by: Pablo Marquez Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/runtime/NEON/functions/NELSTMLayer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/runtime/NEON/functions/NELSTMLayer.cpp b/src/runtime/NEON/functions/NELSTMLayer.cpp index 4d93774536..a0adbb0f43 100644 --- a/src/runtime/NEON/functions/NELSTMLayer.cpp +++ b/src/runtime/NEON/functions/NELSTMLayer.cpp @@ -92,7 +92,7 @@ void NELSTMLayer::configure(const ITensor *input, scratch_buffer->info(), output_state_out->info(), cell_state_out->info(), output->info(), lstm_params_info, activation_info, cell_threshold, projection_threshold)); - const TensorShape cell_state_shape = cell_state_in->info()->tensor_shape(); + const TensorShape cell_state_shape = cell_state_in->info()->tensor_shape(); // Configure block that calculates the forget gate // forget_gate = Activation(input * input_to_forget_weights + output_state_in * recurrent_to_forget_weights + PixelWiseMul(cell_state, cell_to_forget_weights) + forget_gate_bias) @@ -495,6 +495,7 @@ void NELSTMLayer::run() _memory_group.acquire(); + _concat_inputs_forget_gate.run(); _fully_connected_forget_gate.run(); if(_run_peephole_opt) @@ -574,7 +575,6 @@ void NELSTMLayer::prepare() { if(!_is_prepared) { - _concat_inputs_forget_gate.run(); _concat_weights_forget_gate.run(); if(!_run_cifg_opt) { -- cgit v1.2.1