aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/LayerSupport.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/LayerSupport.hpp')
-rw-r--r--include/armnn/LayerSupport.hpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/include/armnn/LayerSupport.hpp b/include/armnn/LayerSupport.hpp
index d9de76f89c..43a5756e4a 100644
--- a/include/armnn/LayerSupport.hpp
+++ b/include/armnn/LayerSupport.hpp
@@ -37,8 +37,10 @@ bool IsConstantSupported(Compute compute,
bool IsConvolution2dSupported(Compute compute,
const TensorInfo& input,
+ const TensorInfo& output,
const Convolution2dDescriptor& descriptor,
const TensorInfo& weights,
+ const TensorInfo& biases,
char* reasonIfUnsupported = nullptr,
size_t reasonIfUnsupportedMaxLength = 1024);