diff options
| -rw-r--r-- | src/runtime/NEON/functions/NELSTMLayer.cpp | 4 |
1 files 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) { |