From 13a2080f9e88dacf0b96744ecf60a5e9debf9010 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 16 Jan 2019 18:21:08 +0000 Subject: COMPMID-1891: Fixes NELSTM padding issues. Change-Id: Ica32ddec92ea13f66ced3386394f3734d832151b Reviewed-on: https://review.mlplatform.org/530 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Matthew Bentham --- src/runtime/NEON/functions/NELSTMLayer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/runtime/NEON') diff --git a/src/runtime/NEON/functions/NELSTMLayer.cpp b/src/runtime/NEON/functions/NELSTMLayer.cpp index 7c7580a6bb..2a2980bebe 100644 --- a/src/runtime/NEON/functions/NELSTMLayer.cpp +++ b/src/runtime/NEON/functions/NELSTMLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -246,7 +246,6 @@ void NELSTMLayer::configure(const ITensor *input, _output1.allocator()->allocate(); } _activation_output.configure(output_gate_out, nullptr, ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::LOGISTIC)); - output_gate_out->allocator()->allocate(); // Configure block that calculates the output state /** lstm_res = PixelwiseMul(output, Activation(cell_state)) @@ -265,6 +264,7 @@ void NELSTMLayer::configure(const ITensor *input, _activation_output_state.configure(&_cell_state_out1, &_cell_state_activation, activation_info); _pixelwise_mul_output_state2.configure(&_cell_state_activation, output_gate_out, output_state_out_tmp, 1, ConvertPolicy::SATURATE, RoundingPolicy::TO_ZERO); _cell_state_activation.allocator()->allocate(); + output_gate_out->allocator()->allocate(); if(lstm_params.has_projection()) { -- cgit v1.2.1