diff options
Diffstat (limited to 'src/backends/cl/workloads/ClSoftmaxBaseWorkload.hpp')
| -rw-r--r-- | src/backends/cl/workloads/ClSoftmaxBaseWorkload.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClSoftmaxBaseWorkload.hpp b/src/backends/cl/workloads/ClSoftmaxBaseWorkload.hpp index b800056cdf..8d73060162 100644 --- a/src/backends/cl/workloads/ClSoftmaxBaseWorkload.hpp +++ b/src/backends/cl/workloads/ClSoftmaxBaseWorkload.hpp @@ -5,6 +5,7 @@ #pragma once +#include <armnn/Descriptors.hpp> #include <armnn/Tensor.hpp> #include <arm_compute/core/Error.h> @@ -12,6 +13,7 @@ namespace armnn { arm_compute::Status ClSoftmaxWorkloadValidate(const TensorInfo& input, - const TensorInfo& output); + const TensorInfo& output, + const SoftmaxDescriptor& descriptor); } // namespace armnn |