From 0cc37c31a36e7b146cf9640ad69925d7c06b71b4 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 14 Nov 2018 15:54:26 +0000 Subject: COMPMID-1451: Fix the shape of scratch_buffer in case of CIFG In case of CIFG optimisation scratch buffer should have a size of [batch_size, num_units * 3] else [batch_size, num_units * 4]. Change-Id: I43e46f7b52e791472f1196f36e9142240ba76c5c --- src/runtime/NEON/functions/NELSTMLayer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/runtime/NEON/functions/NELSTMLayer.cpp') diff --git a/src/runtime/NEON/functions/NELSTMLayer.cpp b/src/runtime/NEON/functions/NELSTMLayer.cpp index 934761a8ef..7c7580a6bb 100644 --- a/src/runtime/NEON/functions/NELSTMLayer.cpp +++ b/src/runtime/NEON/functions/NELSTMLayer.cpp @@ -286,7 +286,7 @@ void NELSTMLayer::configure(const ITensor *input, // Vector for holding the tensors to store in scratch buffer std::vector scratch_inputs; - if(lstm_params.has_cifg_opt()) + if(!lstm_params.has_cifg_opt()) { scratch_inputs.emplace_back(&_input_gate_out1); } @@ -444,7 +444,7 @@ Status NELSTMLayer::validate(const ITensorInfo *input, // Validate scratch concatenation std::vector inputs_vector_info_raw; - if(lstm_params.has_cifg_opt()) + if(!lstm_params.has_cifg_opt()) { inputs_vector_info_raw.push_back(&input_gate); } -- cgit v1.2.1