diff options
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionBaseWorkload.hpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionBaseWorkload.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionBaseWorkload.hpp b/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionBaseWorkload.hpp index a879efc89e..502d570552 100644 --- a/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionBaseWorkload.hpp +++ b/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionBaseWorkload.hpp @@ -7,6 +7,8 @@ #include "backends/ClWorkloadUtils.hpp" +#include <boost/optional.hpp> + namespace armnn { @@ -14,7 +16,7 @@ arm_compute::Status ClDepthwiseConvolutionWorkloadValidate(const TensorInfo& inp const TensorInfo& output, const DepthwiseConvolution2dDescriptor& descriptor, const TensorInfo& weights, - const TensorInfo& biases); + const boost::optional<TensorInfo>& biases); template<armnn::DataType... dataTypes> class ClDepthwiseConvolutionBaseWorkload : public TypedWorkload<DepthwiseConvolution2dQueueDescriptor, dataTypes...> |