diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-03-20 10:30:58 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
| commit | 36a559e49a3d5b832b1cffd47f2298f452616bb9 (patch) | |
| tree | ae09231cbc75bfe27bd1f9a2a8b92386e24fca82 /arm_compute/core/utils | |
| parent | ee33ea5a6e1aa0faac1cc8b5a269bd4f89854821 (diff) | |
| download | ComputeLibrary-36a559e49a3d5b832b1cffd47f2298f452616bb9.tar.gz | |
COMPMID-992 Implement CL RNN function
Change-Id: I8dbada5fabedbb8523e433ba73d504bd15b81466
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/125787
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
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 |