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
3 files changed