diff options
Diffstat (limited to 'src/armnn/backends/NeonLayerSupport.hpp')
-rw-r--r-- | src/armnn/backends/NeonLayerSupport.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/backends/NeonLayerSupport.hpp b/src/armnn/backends/NeonLayerSupport.hpp index 6f9fe9c20e..45032444a4 100644 --- a/src/armnn/backends/NeonLayerSupport.hpp +++ b/src/armnn/backends/NeonLayerSupport.hpp @@ -8,6 +8,8 @@ #include <armnn/Types.hpp> #include <armnn/Tensor.hpp> +#include <boost/optional.hpp> + namespace armnn { @@ -46,7 +48,7 @@ bool IsConvolution2dSupportedNeon(const TensorInfo& input, const TensorInfo& output, const Convolution2dDescriptor& descriptor, const TensorInfo& weights, - const TensorInfo& biases, + const boost::optional<TensorInfo>& biases, std::string* reasonIfUnsupported = nullptr); @@ -54,7 +56,7 @@ bool IsDepthwiseConvolutionSupportedNeon(const TensorInfo& input, const TensorInfo& output, const DepthwiseConvolution2dDescriptor& descriptor, const TensorInfo& weights, - const TensorInfo& biases, + const boost::optional<TensorInfo>& biases, std::string* reasonIfUnsupported = nullptr); bool IsFullyConnectedSupportedNeon(const TensorInfo& input, |