diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp b/src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp index 4b43052365..0fa9d43b15 100644 --- a/src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp +++ b/src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp @@ -7,8 +7,10 @@ #include "NeonWorkloadUtils.hpp" #include <aclCommon/ArmComputeTensorUtils.hpp> + #include <backendsCommon/CpuTensorHandle.hpp> -#include <TensorUtils.hpp> + +#include <armnnUtils/TensorUtils.hpp> #include <arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h> |