diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonConvolution2dWorkload.hpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonConvolution2dWorkload.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/neon/workloads/NeonConvolution2dWorkload.hpp b/src/backends/neon/workloads/NeonConvolution2dWorkload.hpp index 860d78ba7e..4b6e58ce41 100644 --- a/src/backends/neon/workloads/NeonConvolution2dWorkload.hpp +++ b/src/backends/neon/workloads/NeonConvolution2dWorkload.hpp @@ -21,7 +21,8 @@ arm_compute::Status NeonConvolution2dWorkloadValidate(const TensorInfo& input, const Convolution2dDescriptor& descriptor, const TensorInfo& weights, const Optional<TensorInfo>& biases, - bool isFastMathEnabled = false); + bool isFastMathEnabled = false, + const ActivationDescriptor* activationDescriptor = nullptr); class NeonConvolution2dWorkload : public BaseWorkload<Convolution2dQueueDescriptor> { |