diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-01-16 18:21:08 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-01-17 15:11:46 +0000 |
commit | 13a2080f9e88dacf0b96744ecf60a5e9debf9010 (patch) | |
tree | c72cbca896d288eaeb3b54f17fe87ee3a02e554a /src/runtime/NEON | |
parent | 1df9ccaf7af1593793101ec2164fba19cf2d05ef (diff) | |
download | ComputeLibrary-13a2080f9e88dacf0b96744ecf60a5e9debf9010.tar.gz |
COMPMID-1891: Fixes NELSTM padding issues.
Change-Id: Ica32ddec92ea13f66ced3386394f3734d832151b
Reviewed-on: https://review.mlplatform.org/530
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Matthew Bentham <matthew.bentham@arm.com>
Diffstat (limited to 'src/runtime/NEON')
-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 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()) { |