diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2018-01-04 10:34:24 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:43:42 +0000 |
| commit | 00afd11eaa7d408ff873732639c9a724fece9058 (patch) | |
| tree | 37dfbaca825c3363dd2197ea85f99f740748b5b0 /arm_compute | |
| parent | 5237e01c342b9301951a799842e9c48813b66fd4 (diff) | |
| download | ComputeLibrary-00afd11eaa7d408ff873732639c9a724fece9058.tar.gz | |
COMPMID-719: NEPermuteKernel refactoring
Change-Id: I91b43d9706ac3244ce43684967ace0b022d35bad
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/114988
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 6 |
1 files changed, 6 insertions, 0 deletions
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) ] |