diff options
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClConvolution2dBaseWorkload.hpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClConvolution2dBaseWorkload.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClConvolution2dBaseWorkload.hpp b/src/armnn/backends/ClWorkloads/ClConvolution2dBaseWorkload.hpp index c4ef152361..c79e476dd3 100644 --- a/src/armnn/backends/ClWorkloads/ClConvolution2dBaseWorkload.hpp +++ b/src/armnn/backends/ClWorkloads/ClConvolution2dBaseWorkload.hpp @@ -7,6 +7,8 @@ #include "backends/ClWorkloadUtils.hpp" +#include <boost/optional.hpp> + namespace armnn { @@ -14,6 +16,6 @@ arm_compute::Status ClConvolution2dWorkloadValidate(const TensorInfo& input, const TensorInfo& output, const Convolution2dDescriptor& descriptor, const TensorInfo& weights, - const TensorInfo& biases); + const boost::optional<TensorInfo>& biases); } //namespace armnn |