diff options
Diffstat (limited to 'arm_compute/core/utils')
-rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 10 |
1 files changed, 10 insertions, 0 deletions
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(); |