From 00afd11eaa7d408ff873732639c9a724fece9058 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Thu, 4 Jan 2018 10:34:24 +0000 Subject: COMPMID-719: NEPermuteKernel refactoring Change-Id: I91b43d9706ac3244ce43684967ace0b022d35bad Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/114988 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- arm_compute/core/utils/misc/ShapeCalculator.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'arm_compute/core') diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index c7667f2c7b..61834b88a9 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -33,6 +33,12 @@ namespace misc { namespace shape_calculator { +inline TensorShape compute_permutation_output_shape(const ITensorInfo &input, const PermutationVector &perm) +{ + TensorShape output_shape = input.tensor_shape(); + permute(output_shape, perm); + return output_shape; +} inline TensorShape compute_interleaved_shape(const ITensorInfo &a) { // The interleaved output matrix will have the following shape: [ a_height * 4, ceil(a_width / 4.0f) ] -- cgit v1.2.1