diff options
Diffstat (limited to 'utils/GraphUtils.h')
| -rw-r--r-- | utils/GraphUtils.h | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h index 8558b9066c..6d537865ba 100644 --- a/utils/GraphUtils.h +++ b/utils/GraphUtils.h @@ -474,6 +474,24 @@ inline std::unique_ptr<graph::ITensorAccessor> get_npy_output_accessor(const std } } +/** Permutes a given tensor shape given the input and output data layout + * + * @param[in] tensor_shape Tensor shape to permute + * @param[in] in_data_layout Input tensor shape data layout + * @param[in] out_data_layout Output tensor shape data layout + * + * @return Permuted tensor shape + */ +inline TensorShape permute_shape(TensorShape tensor_shape, DataLayout in_data_layout, DataLayout out_data_layout) +{ + if(in_data_layout != out_data_layout) + { + arm_compute::PermutationVector perm_vec = (in_data_layout == DataLayout::NCHW) ? arm_compute::PermutationVector(2U, 0U, 1U) : arm_compute::PermutationVector(1U, 2U, 0U); + arm_compute::permute(tensor_shape, perm_vec); + } + return tensor_shape; +} + /** Utility function to return the TargetHint * * @param[in] target Integer value which expresses the selected target. Must be 0 for NEON or 1 for OpenCL or 2 (OpenCL with Tuner) |