1 files changed, 8 insertions, 0 deletions
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