diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-12-12 11:44:44 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:33 +0000 |
commit | 0223a78ba43f0b106347108857d2f8cbfe857198 (patch) | |
tree | e6751b5418534595d890f401929284fe6d198f4b /arm_compute/runtime | |
parent | 1568621e07cef67c5bb01fa4cc827e218302040c (diff) | |
download | ComputeLibrary-0223a78ba43f0b106347108857d2f8cbfe857198.tar.gz |
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>
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h index 6d8ce1f6f4..09a54968bb 100644 --- a/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h @@ -62,7 +62,7 @@ public: * 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 @@ public: * 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 @@ private: NEDirectConvolutionLayerKernel _conv_kernel; NEFillBorderKernel _input_border_handler; Tensor _accumulator; + bool _has_bias; }; } #endif /* __ARM_COMPUTE_NEDIRECTCONVOLUTIONLAYER_H__ */ |