aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/NeonLayerSupport.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/backends/NeonLayerSupport.hpp')
-rw-r--r--src/armnn/backends/NeonLayerSupport.hpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/backends/NeonLayerSupport.hpp b/src/armnn/backends/NeonLayerSupport.hpp
index b2ac49ae0d..ce2ecec459 100644
--- a/src/armnn/backends/NeonLayerSupport.hpp
+++ b/src/armnn/backends/NeonLayerSupport.hpp
@@ -39,8 +39,10 @@ bool IsConstantSupportedNeon(const TensorInfo& output,
std::string* reasonIfUnsupported = nullptr);
bool IsConvolution2dSupportedNeon(const TensorInfo& input,
+ const TensorInfo& output,
const Convolution2dDescriptor& descriptor,
const TensorInfo& weights,
+ const TensorInfo& biases,
std::string* reasonIfUnsupported = nullptr);
bool IsDepthwiseConvolutionSupportedNeon(const TensorInfo& input,