COMPMID-556: Fix bugs around NEDirectConvolutionLayer
Change-Id: Ib4af25cd6dae78ed4ec89f4272cfaa2356359446
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/112867
Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h
index 6d8ce1f..09a5496 100644
--- a/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h
@@ -62,7 +62,7 @@
* Supported sizes: 1x1, 3x3 and 5x5.
* The 3rd dimension must be the same as the input's volume 3rd dimension.
* Data type supported: Same as @p input.
- * @param[in] bias Set of biases. Data type supported: Same as @p input.
+ * @param[in] bias Set of biases. Can be nullptr. Data type supported: Same as @p input.
* @param[out] output Output tensor.
* The 3rd dimensions must be equal to the 4th dimension of the @p kernels tensor. Data types supported: Same as @p input.
* @param[in] conv_info Contains padding and stride information described in @ref PadStrideInfo.
@@ -80,7 +80,7 @@
* Supported sizes: 1x1, 3x3 and 5x5.
* The 3rd dimension must be the same as the input's volume 3rd dimension.
* Data type supported: Same as @p input.
- * @param[in] bias Set of biases. Data type supported: Same as @p input.
+ * @param[in] bias Set of biases. Can be nullptr. Data type supported: Same as @p input.
* @param[in] output Output tensor.
* The 3rd dimensions must be equal to the 4th dimension of the @p kernels tensor. Data types supported: Same as @p input.
* @param[in] conv_info Contains padding and stride information described in @ref PadStrideInfo.
@@ -98,6 +98,7 @@
NEDirectConvolutionLayerKernel _conv_kernel;
NEFillBorderKernel _input_border_handler;
Tensor _accumulator;
+ bool _has_bias;
};
}
#endif /* __ARM_COMPUTE_NEDIRECTCONVOLUTIONLAYER_H__ */