diff options
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClActivationFloat32Workload.hpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClActivationFloat32Workload.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClActivationFloat32Workload.hpp b/src/armnn/backends/ClWorkloads/ClActivationFloat32Workload.hpp index 9bab4202be..9fbfe95856 100644 --- a/src/armnn/backends/ClWorkloads/ClActivationFloat32Workload.hpp +++ b/src/armnn/backends/ClWorkloads/ClActivationFloat32Workload.hpp @@ -9,9 +9,12 @@ namespace armnn { +arm_compute::Status ClActivationWorkloadValidate(const TensorInfo& input, + const TensorInfo& output, + const ActivationDescriptor& descriptor); -// Activation layer execution -class ClActivationFloat32Workload : public Float32Workload<ActivationQueueDescriptor> +// Activation layer execution. +class ClActivationFloat32Workload : public FloatWorkload<ActivationQueueDescriptor> { public: ClActivationFloat32Workload(const ActivationQueueDescriptor& descriptor, const WorkloadInfo& info); |