diff options
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp b/src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp index 751d8ff1a4..b800056cdf 100644 --- a/src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp +++ b/src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp @@ -5,7 +5,8 @@ #pragma once -#include "backends/ClWorkloadUtils.hpp" +#include <armnn/Tensor.hpp> +#include <arm_compute/core/Error.h> namespace armnn { |