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 --- tests/validation/fixtures/LSTMLayerFixture.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tests/validation/fixtures') diff --git a/tests/validation/fixtures/LSTMLayerFixture.h b/tests/validation/fixtures/LSTMLayerFixture.h index bc892bfecf..7f11265932 100644 --- a/tests/validation/fixtures/LSTMLayerFixture.h +++ b/tests/validation/fixtures/LSTMLayerFixture.h @@ -100,7 +100,7 @@ protected: TensorType projection_w; TensorType projection_bias; - bool cifg_opt = scratch_shape.x() == cell_bias_shape.x() * 4 ? true : false; + bool cifg_opt = scratch_shape.x() == cell_bias_shape.x() * 4 ? false : true; FunctionParams lstm_params; -- cgit v1.2.1