aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClMultiplicationWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/cl/workloads/ClMultiplicationWorkload.hpp')
-rw-r--r--src/backends/cl/workloads/ClMultiplicationWorkload.hpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClMultiplicationWorkload.hpp b/src/backends/cl/workloads/ClMultiplicationWorkload.hpp
index 732bb16dcc..461449cc35 100644
--- a/src/backends/cl/workloads/ClMultiplicationWorkload.hpp
+++ b/src/backends/cl/workloads/ClMultiplicationWorkload.hpp
@@ -14,7 +14,8 @@ namespace armnn
arm_compute::Status ClMultiplicationWorkloadValidate(const TensorInfo& input0,
const TensorInfo& input1,
- const TensorInfo& output);
+ const TensorInfo& output,
+ const ActivationDescriptor* activationDescriptor = nullptr);
class ClMultiplicationWorkload : public BaseWorkload<MultiplicationQueueDescriptor>
{