diff options
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClSoftmaxFloatWorkload.hpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClSoftmaxFloatWorkload.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClSoftmaxFloatWorkload.hpp b/src/armnn/backends/ClWorkloads/ClSoftmaxFloatWorkload.hpp index de8ff7e17e..965b845cf8 100644 --- a/src/armnn/backends/ClWorkloads/ClSoftmaxFloatWorkload.hpp +++ b/src/armnn/backends/ClWorkloads/ClSoftmaxFloatWorkload.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> |