diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp b/src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp index be26359662..c915555dd7 100644 --- a/src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp +++ b/src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp @@ -5,11 +5,18 @@ #include "NeonDepthwiseConvolutionWorkload.hpp" +#include "NeonWorkloadUtils.hpp" + +#include <DataLayoutIndexed.hpp> #include <aclCommon/ArmComputeTensorUtils.hpp> #include <neon/NeonLayerSupport.hpp> #include <backendsCommon/CpuTensorHandle.hpp> #include <backendsCommon/WorkloadUtils.hpp> +#include <arm_compute/runtime/NEON/functions/NEDepthwiseConvolutionLayer.h> + +using namespace armnnUtils; + namespace armnn { |