COMPMID-3045: CTS failures in ARGMAX/MIN
Change-Id: I35276a3d95dc99a7f4dea00e89c8ed206a5f13f1
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2669
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
diff --git a/tests/validation/CL/ArgMinMax.cpp b/tests/validation/CL/ArgMinMax.cpp
index 4bee942..e5decb8 100644
--- a/tests/validation/CL/ArgMinMax.cpp
+++ b/tests/validation/CL/ArgMinMax.cpp
@@ -47,12 +47,14 @@
const auto ArgMinMaxSmallDataset = framework::dataset::make("Shape",
{
TensorShape{ 2U, 7U, 1U, 3U },
- TensorShape{ 128U, 64U, 21U, 3U },
+ TensorShape{ 149U, 5U, 1U, 2U },
+ TensorShape{ 166U, 5U, 1U, 2U },
+ TensorShape{ 322U, 5U, 1U, 2U },
+ TensorShape{ 128U, 5U, 21U, 3U },
TensorShape{ 2560, 2U, 2U, 2U },
});
-const auto ArgMinMaxLargeDataset = framework::dataset::make("Shape",
-{ TensorShape{ 517U, 123U, 13U, 2U } });
+const auto ArgMinMaxLargeDataset = framework::dataset::make("Shape", { TensorShape{ 517U, 123U, 13U, 2U } });
} // namespace
TEST_SUITE(CL)
TEST_SUITE(ArgMinMax)