commit 4f8ce13e287867a699b4020583173339570e1917
author Giorgio Arena <giorgio.arena@arm.com> Thu Jan 07 16:35:05 2021 +0000
committer Giorgio Arena <giorgio.arena@arm.com> Fri Jan 08 17:11:46 2021 +0000
tree 5d10a5196afc99a5ac516a5793135de524b1df40
parent 70eb53b09adf57cc6ff7435c795e65a4f4697f67

[Nightly Failure] Fix NEIm2Col argument validation errors

- Permute input shape if NHWC for Im2Col's fixture

Resolves: COMPMID-4081

Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: I9a46569f1cb2c09932ffb1b0afda6d66e93452de
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4777
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>

tests/validation/fixtures/Im2ColFixture.h

diff --git a/tests/validation/fixtures/Im2ColFixture.h b/tests/validation/fixtures/Im2ColFixture.h
index e1f33a3..b6cf18b 100644
--- a/tests/validation/fixtures/Im2ColFixture.h
+++ b/tests/validation/fixtures/Im2ColFixture.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2017-2020 Arm Limited.
+ * Copyright (c) 2017-2021 Arm Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -59,6 +59,11 @@
         _has_bias    = data_type != DataType::QASYMM8;
         _num_groups  = num_groups;
 
+        if(_data_layout == DataLayout::NHWC)
+        {
+            permute(input_shape, PermutationVector(2U, 0U, 1U));
+        }
+
         TensorInfo input_info(input_shape, 1, data_type);
         input_info.set_data_layout(_data_layout);