diff options
| author | Manuel Bottini <manuel.bottini@arm.com> | 2019-06-20 16:00:27 +0100 |
|---|---|---|
| committer | Manuel Bottini <manuel.bottini@arm.com> | 2019-07-11 16:52:18 +0000 |
| commit | bffb41e06c1276af00e1605ef934d05fa61f7127 (patch) | |
| tree | 7c9cfe90e82a8107ad8e32272c4e40c4b63182ef /arm_compute/graph/backends/FusedConvolutionBatchNormalizationFunction.h | |
| parent | c1b76faf6be5c33dbf3269faea95e185ac37992f (diff) | |
| download | ComputeLibrary-bffb41e06c1276af00e1605ef934d05fa61f7127.tar.gz | |
COMPMID-2273: Fuse Batch Normalization with Depthwise Convolution layer at graph level (only for CL)
Change-Id: I1d941c6e66722f39583bf68148c980bb28ff89a1
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1423
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/graph/backends/FusedConvolutionBatchNormalizationFunction.h')
| -rw-r--r-- | arm_compute/graph/backends/FusedConvolutionBatchNormalizationFunction.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/arm_compute/graph/backends/FusedConvolutionBatchNormalizationFunction.h b/arm_compute/graph/backends/FusedConvolutionBatchNormalizationFunction.h index 92af17b227..a6da76bb06 100644 --- a/arm_compute/graph/backends/FusedConvolutionBatchNormalizationFunction.h +++ b/arm_compute/graph/backends/FusedConvolutionBatchNormalizationFunction.h @@ -54,7 +54,7 @@ public: * Data types supported: QASYMM8/F16/F32. * @param[in] weights Weights tensor. Weights are 4D tensor with dimensions [kernel_x, kernel_y, IFM, OFM]. Data type supported: Same as @p input. * @param[in] bias Biases tensor. Shared biases supported. Biases are 1D tensor with dimensions [OFM]. - * Data type supported: Should match @p input data type, except for input of QASYMM8 type where biases should be of S32 type. + * Data type supported: Should match @p input data type. * @param[out] output Destination tensor. 3 lower dimensions represent a single output [width, height, OFM], while the rest represent batch of outputs. * Data types supported: Same as @p input. * @param[in] mean Mean values tensor. 1 dimension with size equal to the feature maps [FM]. Data types supported: Same as @p input |