From 36a559e49a3d5b832b1cffd47f2298f452616bb9 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Tue, 20 Mar 2018 10:30:58 +0000 Subject: COMPMID-992 Implement CL RNN function Change-Id: I8dbada5fabedbb8523e433ba73d504bd15b81466 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/125787 Reviewed-by: Georgios Pinitas Tested-by: Jenkins --- arm_compute/core/utils/misc/ShapeCalculator.h | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'arm_compute/core/utils') diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index 383fc6cda6..8816819bcd 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -282,6 +282,14 @@ inline TensorShape compute_min_max_shape(const ITensorInfo *input) return output_shape; } +inline TensorShape compute_rnn_shape(const ITensorInfo *input, const unsigned int batch_size) +{ + TensorShape output_shape{ input->tensor_shape() }; + output_shape.set(1, batch_size); + + return output_shape; +} + } // namespace shape_calculator } // namespace misc } // namespace arm_compute -- cgit v1.2.1