diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/datasets/ShapeDatasets.h | 6 | ||||
-rw-r--r-- | tests/validation/NEON/ConvolutionLayer.cpp | 2 |
2 files changed, 4 insertions, 4 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 733394224c..4b563708e1 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -100,7 +100,8 @@ public: Small3DShapes() : ShapeDataset("Shape", { - TensorShape{ 7U, 7U, 5U }, + TensorShape{ 1U, 7U, 7U }, + TensorShape{ 7U, 7U, 5U }, TensorShape{ 27U, 13U, 37U }, TensorShape{ 128U, 64U, 21U } }) @@ -128,7 +129,8 @@ public: Small4DShapes() : ShapeDataset("Shape", { - TensorShape{ 7U, 7U, 5U, 3U }, + TensorShape{ 1U, 7U, 1U, 3U }, + TensorShape{ 7U, 7U, 5U, 3U }, TensorShape{ 27U, 13U, 37U, 2U }, TensorShape{ 128U, 64U, 21U, 3U } }) diff --git a/tests/validation/NEON/ConvolutionLayer.cpp b/tests/validation/NEON/ConvolutionLayer.cpp index f6736785bd..ed4f160a37 100644 --- a/tests/validation/NEON/ConvolutionLayer.cpp +++ b/tests/validation/NEON/ConvolutionLayer.cpp @@ -68,7 +68,6 @@ const auto CNNDataTypes = framework::dataset::make("DataType", TEST_SUITE(NEON) -#if defined(__aarch64__) TEST_SUITE(WinogradLayer) template <typename T> using NEWinogradLayerFixture = WinogradLayerValidationFixture<Tensor, Accessor, NEWinogradLayer, T>; @@ -82,7 +81,6 @@ FIXTURE_DATA_TEST_CASE(RunSmall, NEWinogradLayerFixture<float>, framework::Datas TEST_SUITE_END() TEST_SUITE_END() -#endif /* __aarch64__ */ TEST_SUITE(ConvolutionLayer) |