diff options
Diffstat (limited to 'src/backends')
| -rw-r--r-- | src/backends/aclCommon/ArmComputeUtils.hpp | 2 | ||||
| -rw-r--r-- | src/backends/cl/workloads/ClBatchMatMulWorkload.cpp | 1 | ||||
| -rw-r--r-- | src/backends/neon/workloads/NeonBatchMatMulWorkload.cpp | 2 |
3 files changed, 4 insertions, 1 deletions
diff --git a/src/backends/aclCommon/ArmComputeUtils.hpp b/src/backends/aclCommon/ArmComputeUtils.hpp index fab52ffb0f..d9b93f886c 100644 --- a/src/backends/aclCommon/ArmComputeUtils.hpp +++ b/src/backends/aclCommon/ArmComputeUtils.hpp @@ -10,8 +10,8 @@ #include <armnn/utility/NumericCast.hpp> #include <armnn/backends/WorkloadData.hpp> -#include <arm_compute/core/Types.h> #include <arm_compute/runtime/FunctionDescriptors.h> +#include <arm_compute/core/FullyConnectedLayerInfo.h> #if defined(ARMCOMPUTENEON_ENABLED) #include "neon/workloads/NeonReduceWorkload.hpp" diff --git a/src/backends/cl/workloads/ClBatchMatMulWorkload.cpp b/src/backends/cl/workloads/ClBatchMatMulWorkload.cpp index bd0fd51617..cfed1b83a1 100644 --- a/src/backends/cl/workloads/ClBatchMatMulWorkload.cpp +++ b/src/backends/cl/workloads/ClBatchMatMulWorkload.cpp @@ -16,6 +16,7 @@ #include <cl/ClTensorHandle.hpp> +#include <arm_compute/core/MatMulInfo.h> namespace armnn { 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, |