diff --git a/utils/Utils.h b/utils/Utils.h
index adb0e54..0bbdcc2 100644
--- a/utils/Utils.h
+++ b/utils/Utils.h
@@ -393,21 +393,22 @@
                     }
                 }
             }
+
+            TensorShape                    permuted_shape = tensor.info()->tensor_shape();
+            arm_compute::PermutationVector perm;
+            if(are_layouts_different && tensor.info()->tensor_shape().num_dimensions() > 2)
+            {
+                perm                                    = (tensor.info()->data_layout() == arm_compute::DataLayout::NHWC) ? arm_compute::PermutationVector(2U, 0U, 1U) : arm_compute::PermutationVector(1U, 2U, 0U);
+                arm_compute::PermutationVector perm_vec = (tensor.info()->data_layout() == arm_compute::DataLayout::NCHW) ? arm_compute::PermutationVector(2U, 0U, 1U) : arm_compute::PermutationVector(1U, 2U, 0U);
+
+                arm_compute::permute(permuted_shape, perm_vec);
+            }
+
             // Validate tensor shape
             ARM_COMPUTE_ERROR_ON_MSG(_shape.size() != tensor.info()->tensor_shape().num_dimensions(), "Tensor ranks mismatch");
-            if(!_fortran_order)
+            for(size_t i = 0; i < _shape.size(); ++i)
             {
-                for(size_t i = 0; i < _shape.size(); ++i)
-                {
-                    ARM_COMPUTE_ERROR_ON_MSG(tensor.info()->tensor_shape()[i] != _shape[i], "Tensor dimensions mismatch");
-                }
-            }
-            else
-            {
-                for(size_t i = 0; i < _shape.size(); ++i)
-                {
-                    ARM_COMPUTE_ERROR_ON_MSG(tensor.info()->tensor_shape()[i] != _shape[_shape.size() - i - 1], "Tensor dimensions mismatch");
-                }
+                ARM_COMPUTE_ERROR_ON_MSG(permuted_shape[i] != _shape[i], "Tensor dimensions mismatch");
             }
 
             switch(tensor.info()->data_type())
@@ -423,48 +424,31 @@
                     else
                     {
                         // If tensor has padding or is in fortran order accessing tensor elements through execution window.
-                        Window                         window;
-                        TensorShape                    permuted_shape = tensor.info()->tensor_shape();
-                        const unsigned int             num_dims       = _shape.size();
-                        arm_compute::PermutationVector perm;
+                        Window             window;
+                        const unsigned int num_dims = _shape.size();
                         if(_fortran_order)
                         {
                             for(unsigned int dim = 0; dim < num_dims; dim++)
                             {
-                                permuted_shape.set(dim, _shape[dim]);
+                                permuted_shape.set(dim, _shape[num_dims - dim - 1]);
                                 perm.set(dim, num_dims - dim - 1);
                             }
-                        }
-                        else
-                        {
-                            for(unsigned int dim = 0; dim < num_dims; dim++)
+                            if(are_layouts_different)
                             {
-                                perm.set(dim, dim);
-                            }
-                        }
-                        if(are_layouts_different)
-                        {
-                            // Permute only if num_dimensions greater than 2
-                            if(num_dims > 2)
-                            {
-                                if(_file_layout == DataLayout::NHWC) // i.e destination is NCHW --> permute(1,2,0)
+                                // Permute only if num_dimensions greater than 2
+                                if(num_dims > 2)
                                 {
-                                    size_t perm_0 = perm[0];
-                                    perm[0]       = perm[1];
-                                    perm[1]       = perm[2];
-                                    perm[2]       = perm_0;
-                                }
-                                else
-                                {
-                                    // destination layout is NHWC --> permute (2,0,1)
-                                    size_t perm_0 = perm[0];
-                                    perm[0]       = perm[2];
-                                    perm[2]       = perm[1];
-                                    perm[1]       = perm_0;
+                                    if(_file_layout == DataLayout::NHWC) // i.e destination is NCHW --> permute(1,2,0)
+                                    {
+                                        arm_compute::permute(perm, arm_compute::PermutationVector(1U, 2U, 0U));
+                                    }
+                                    else
+                                    {
+                                        arm_compute::permute(perm, arm_compute::PermutationVector(2U, 0U, 1U));
+                                    }
                                 }
                             }
                         }
-
                         window.use_tensor_dimensions(permuted_shape);
 
                         execute_window_loop(window, [&](const Coordinates & id)