diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonConvolution2dWorkload.hpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonConvolution2dWorkload.hpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/backends/neon/workloads/NeonConvolution2dWorkload.hpp b/src/backends/neon/workloads/NeonConvolution2dWorkload.hpp index 3fb408dbaa..54e08a2042 100644 --- a/src/backends/neon/workloads/NeonConvolution2dWorkload.hpp +++ b/src/backends/neon/workloads/NeonConvolution2dWorkload.hpp @@ -17,10 +17,11 @@ namespace armnn { arm_compute::Status NeonConvolution2dWorkloadValidate(const TensorInfo& input, - const TensorInfo& output, - const Convolution2dDescriptor& descriptor, - const TensorInfo& weights, - const Optional<TensorInfo>& biases); + const TensorInfo& output, + const Convolution2dDescriptor& descriptor, + const TensorInfo& weights, + const Optional<TensorInfo>& biases, + bool isFastMathEnabled = false); class NeonConvolution2dWorkload : public BaseWorkload<Convolution2dQueueDescriptor> { |