diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp b/src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp index 434de874ae..b229bc48a2 100644 --- a/src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp +++ b/src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp @@ -7,6 +7,8 @@ #include <aclCommon/ArmComputeTensorUtils.hpp> +#include <arm_compute/runtime/NEON/functions/NESoftmaxLayer.h> + namespace armnn { |