diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonConvolution2dWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonConvolution2dWorkload.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp b/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp index 151132f04d..1080f320e7 100644 --- a/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp +++ b/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp @@ -7,7 +7,9 @@ #include <backendsCommon/CpuTensorHandle.hpp> #include <aclCommon/ArmComputeTensorUtils.hpp> -#include <neon/NeonLayerSupport.hpp> +#include <neon/workloads/NeonWorkloadUtils.hpp> + +#include <arm_compute/runtime/NEON/functions/NEConvolutionLayer.h> #include <armnn/Types.hpp> #include <Half.hpp> |