diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/validation/CL/GenerateProposalsLayer.cpp | 5 | ||||
| -rw-r--r-- | tests/validation/NEON/GenerateProposalsLayer.cpp | 5 |
2 files changed, 8 insertions, 2 deletions
diff --git a/tests/validation/CL/GenerateProposalsLayer.cpp b/tests/validation/CL/GenerateProposalsLayer.cpp index d0ceef5106..646a0e0530 100644 --- a/tests/validation/CL/GenerateProposalsLayer.cpp +++ b/tests/validation/CL/GenerateProposalsLayer.cpp @@ -22,12 +22,13 @@ * SOFTWARE. */ #include "arm_compute/runtime/CL/CLScheduler.h" -#include "arm_compute/runtime/CL/functions/CLComputeAllAnchors.h" #include "arm_compute/runtime/CL/functions/CLGenerateProposalsLayer.h" #include "arm_compute/runtime/CL/functions/CLPermute.h" #include "arm_compute/runtime/CL/functions/CLSlice.h" +#include "src/core/CL/kernels/CLGenerateProposalsLayerKernel.h" #include "tests/CL/CLAccessor.h" #include "tests/CL/CLArrayAccessor.h" +#include "tests/CL/Helper.h" #include "tests/Globals.h" #include "tests/framework/Macros.h" #include "tests/framework/datasets/Datasets.h" @@ -43,6 +44,8 @@ namespace validation { namespace { +using CLComputeAllAnchors = CLSynthetizeFunction<CLComputeAllAnchorsKernel>; + template <typename U, typename T> inline void fill_tensor(U &&tensor, const std::vector<T> &v) { diff --git a/tests/validation/NEON/GenerateProposalsLayer.cpp b/tests/validation/NEON/GenerateProposalsLayer.cpp index d14b8cb8c4..6a28fecf7f 100644 --- a/tests/validation/NEON/GenerateProposalsLayer.cpp +++ b/tests/validation/NEON/GenerateProposalsLayer.cpp @@ -22,13 +22,14 @@ * SOFTWARE. */ #include "arm_compute/runtime/NEON/NEScheduler.h" -#include "arm_compute/runtime/NEON/functions/NEComputeAllAnchors.h" #include "arm_compute/runtime/NEON/functions/NEGenerateProposalsLayer.h" #include "arm_compute/runtime/NEON/functions/NEPermute.h" #include "arm_compute/runtime/NEON/functions/NESlice.h" +#include "src/core/NEON/kernels/NEGenerateProposalsLayerKernel.h" #include "tests/Globals.h" #include "tests/NEON/Accessor.h" #include "tests/NEON/ArrayAccessor.h" +#include "tests/NEON/Helper.h" #include "tests/framework/Macros.h" #include "tests/framework/datasets/Datasets.h" #include "tests/validation/Validation.h" @@ -43,6 +44,8 @@ namespace validation { namespace { +using NEComputeAllAnchors = NESynthetizeFunction<NEComputeAllAnchorsKernel>; + template <typename U, typename T> inline void fill_tensor(U &&tensor, const std::vector<T> &v) { |