commit 57aa98e078fb665e5567e0aef806304f07780077
author Michalis Spyrou <michalis.spyrou@arm.com> Thu May 30 13:31:57 2019 +0100
committer Michalis Spyrou <michalis.spyrou@arm.com> Thu May 30 14:31:58 2019 +0000
tree d1cd68bd04cf18b52185126b32256bd4acd2e4f7
parent db18a6f67c32d8d01830186529c3cc61741385cc

COMPMID-2377: (Nightly) NEON SpaceToDepthLayer errors

Change-Id: I5f76ea3ba66cf00c4d4de22d6700358c982f603d
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1256
Reviewed-by: Manuel Bottini <manuel.bottini@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

tests/datasets/SpaceToDepthDataset.h

diff --git a/tests/datasets/SpaceToDepthDataset.h b/tests/datasets/SpaceToDepthDataset.h
index 563a475..eb81c9a 100644
--- a/tests/datasets/SpaceToDepthDataset.h
+++ b/tests/datasets/SpaceToDepthDataset.h
@@ -126,7 +126,7 @@
     LargeSpaceToDepthLayerDataset()
     {
         add_config(TensorShape(128U, 64U, 2U, 1U), TensorShape(64U, 32U, 8U, 1U), 2);
-        add_config(TensorShape(512U, 64U, 2U, 1U), TensorShape(128U, 16U, 8U, 4U), 2);
+        add_config(TensorShape(512U, 64U, 2U, 4U), TensorShape(256U, 32U, 8U, 4U), 2);
     }
 };
 } // namespace datasets