aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/NEON/Col2Im.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/NEON/Col2Im.cpp')
-rw-r--r--tests/validation/NEON/Col2Im.cpp7
1 files changed, 5 insertions, 2 deletions
diff --git a/tests/validation/NEON/Col2Im.cpp b/tests/validation/NEON/Col2Im.cpp
index 5f98b455d9..9139f0cca8 100644
--- a/tests/validation/NEON/Col2Im.cpp
+++ b/tests/validation/NEON/Col2Im.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017 Arm Limited.
+ * Copyright (c) 2017-2020 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -22,7 +22,8 @@
* SOFTWARE.
*/
#include "arm_compute/core/Types.h"
-#include "arm_compute/runtime/NEON/functions/NECol2Im.h"
+#include "src/core/NEON/kernels/NECol2ImKernel.h"
+#include "tests/NEON/Helper.h"
#include "tests/datasets/ShapeDatasets.h"
#include "tests/framework/Asserts.h"
#include "tests/framework/Macros.h"
@@ -38,6 +39,8 @@ namespace validation
TEST_SUITE(NEON)
TEST_SUITE(Col2Im)
+using NECol2Im = NESynthetizeFunction<NECol2ImKernel>;
+
// *INDENT-OFF*
// clang-format off
DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip(zip(