From 3175fcf63249673f33fd1638879adad4baab545b Mon Sep 17 00:00:00 2001 From: giuros01 Date: Wed, 21 Nov 2018 09:59:17 +0000 Subject: COMPMID-1720: CL: Implement Tile Change-Id: I2a18f0acea382960a8bc71a8f56928a5998f0dd6 --- arm_compute/core/utils/misc/ShapeCalculator.h | 10 ++++++++++ 1 file changed, 10 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 da9ff56fd0..38906dfc9b 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -671,6 +671,16 @@ inline TensorShape compute_padded_shape(const TensorShape &input_shape, const Pa return padded_shape; } +inline TensorShape compute_tiled_shape(const TensorShape &input_shape, const Multiples &multiples) +{ + TensorShape tiled_shape = input_shape; + for(size_t dim = 0; dim < multiples.size(); ++dim) + { + tiled_shape.set(dim, input_shape[dim] * multiples[dim]); + } + return tiled_shape; +} + inline TensorShape compute_upsample_shape(const ITensorInfo &input, const Size2D &info) { const DataLayout data_layout = input.data_layout(); -- cgit v1.2.1