aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonBatchMatMulWorkload.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon/workloads/NeonBatchMatMulWorkload.cpp')
-rw-r--r--src/backends/neon/workloads/NeonBatchMatMulWorkload.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonBatchMatMulWorkload.cpp b/src/backends/neon/workloads/NeonBatchMatMulWorkload.cpp
index 9b22033bd1..628e314046 100644
--- a/src/backends/neon/workloads/NeonBatchMatMulWorkload.cpp
+++ b/src/backends/neon/workloads/NeonBatchMatMulWorkload.cpp
@@ -12,6 +12,8 @@
#include <backendsCommon/WorkloadUtils.hpp>
+#include <arm_compute/core/MatMulInfo.h>
+
namespace armnn
{
arm_compute::Status NeonBatchMatMulValidate(const TensorInfo& inputInfoX,