IVGCVSW-1324: Make biases truly optional for Conv2d and DepthwiseConvolution

!android-nn-driver:145961

Change-Id: I039ab0adc61725859514246365b5e5b5fa603eaa

1 files changed, 4 insertions, 2 deletions

diff --git a/include/armnn/LayerSupport.hpp b/include/armnn/LayerSupport.hpp
index c875619949..26cbda47e2 100644
--- a/include/armnn/LayerSupport.hpp
+++ b/include/armnn/LayerSupport.hpp
@@ -8,6 +8,8 @@
 #include <armnn/Types.hpp>
 #include <armnn/Tensor.hpp>
 
+#include <boost/optional.hpp>
+
 namespace armnn
 {
 
@@ -58,7 +60,7 @@ bool IsConvolution2dSupported(Compute compute,
                               const TensorInfo& output,
                               const Convolution2dDescriptor& descriptor,
                               const TensorInfo& weights,
-                              const TensorInfo& biases,
+                              const boost::optional<TensorInfo>& biases,
                               char* reasonIfUnsupported = nullptr,
                               size_t reasonIfUnsupportedMaxLength = 1024);
 
@@ -67,7 +69,7 @@ bool IsDepthwiseConvolutionSupported(Compute compute,
                                      const TensorInfo& output,
                                      const DepthwiseConvolution2dDescriptor& descriptor,
                                      const TensorInfo& weights,
-                                     const TensorInfo& biases,
+                                     const boost::optional<TensorInfo>& biases,
                                      char* reasonIfUnsupported = nullptr,
                                      size_t reasonIfUnsupportedMaxLength = 1024);