diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-02-08 16:56:00 +0000 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-02-11 12:18:22 +0000 |
| commit | 6844b8800128d27b469ae4905756deea90dd791c (patch) | |
| tree | 53388b27572898c0809d0fb0cb19179b67bc5f7b | |
| parent | d1fe4ce1290c32d9ebc4c97d81d469fb7546d682 (diff) | |
| download | ComputeLibrary-6844b8800128d27b469ae4905756deea90dd791c.tar.gz | |
COMPMID-1710 Fix NEON LSTM invalid allocation
Change-Id: I5f9d435df63f351062c7091aa4d1d3910a01743e
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/653
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
| -rw-r--r-- | src/runtime/NEON/functions/NELSTMLayer.cpp | 1 |
1 files changed, 0 insertions, 1 deletions
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 |