COMPMID-1386: Add support for converting weights for CL.

Change-Id: I62e3ead903366baeeb1488f233a9b8b0c388c9de
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/140403
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>

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)