diff options
Diffstat (limited to 'arm_compute/core/utils')
-rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 8 |
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 |