diff options
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.hpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.hpp b/src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.hpp index f5eba719bb..29427a5976 100644 --- a/src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.hpp +++ b/src/armnn/backends/ClWorkloads/ClSoftmaxUint8Workload.hpp @@ -5,8 +5,9 @@ #pragma once -#include "backends/ClWorkloadUtils.hpp" +#include "backends/Workload.hpp" +#include <arm_compute/runtime/CL/CLFunctions.h> #include "arm_compute/runtime/MemoryManagerOnDemand.h" #include <memory> |