diff options
Diffstat (limited to 'src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionBaseWorkload.hpp')
-rw-r--r-- | src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionBaseWorkload.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionBaseWorkload.hpp b/src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionBaseWorkload.hpp index 0cead354f8..b19e77ef95 100644 --- a/src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionBaseWorkload.hpp +++ b/src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionBaseWorkload.hpp @@ -7,6 +7,8 @@ #include "backends/NeonWorkloadUtils.hpp" +#include <boost/optional.hpp> + namespace armnn { @@ -14,6 +16,6 @@ arm_compute::Status NeonDepthwiseConvolutionWorkloadValidate(const TensorInfo& i const TensorInfo& output, const DepthwiseConvolution2dDescriptor& descriptor, const TensorInfo& weights, - const TensorInfo& biases); + const boost::optional<TensorInfo>& biases); } // namespace armnn |