diff options
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEReorderLayer.h | 2 | ||||
-rw-r--r-- | src/runtime/NEON/functions/NEReorderLayer.cpp | 1 | ||||
-rw-r--r-- | tests/validation/NEON/ReorderLayer.cpp | 1 |
3 files changed, 3 insertions, 1 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEReorderLayer.h b/arm_compute/runtime/NEON/functions/NEReorderLayer.h index b8d692ac92..a9ce8e3e69 100644 --- a/arm_compute/runtime/NEON/functions/NEReorderLayer.h +++ b/arm_compute/runtime/NEON/functions/NEReorderLayer.h @@ -28,12 +28,12 @@ #include "arm_compute/core/Types.h" #include "arm_compute/runtime/IFunction.h" -#include "src/core/NEON/kernels/NEReorderKernel.h" namespace arm_compute { class ITensor; class ITensorInfo; +class NEReorderKernel; /** Function to compute blocked reorder. */ class NEReorderLayer : public IFunction { diff --git a/src/runtime/NEON/functions/NEReorderLayer.cpp b/src/runtime/NEON/functions/NEReorderLayer.cpp index 53a7718902..2ab1029f02 100644 --- a/src/runtime/NEON/functions/NEReorderLayer.cpp +++ b/src/runtime/NEON/functions/NEReorderLayer.cpp @@ -25,6 +25,7 @@ #include "arm_compute/runtime/NEON/NEScheduler.h" #include "arm_compute/runtime/NEON/functions/NEReorderLayer.h" +#include "src/core/NEON/kernels/NEReorderKernel.h" namespace arm_compute { diff --git a/tests/validation/NEON/ReorderLayer.cpp b/tests/validation/NEON/ReorderLayer.cpp index cc340e9cb1..42fa0f8b00 100644 --- a/tests/validation/NEON/ReorderLayer.cpp +++ b/tests/validation/NEON/ReorderLayer.cpp @@ -32,6 +32,7 @@ #include "tests/framework/datasets/Datasets.h" #include "tests/validation/Validation.h" #include "tests/validation/fixtures/ReorderFixture.h" +#include "src/core/NEON/kernels/NEReorderKernel.h" namespace arm_compute { |