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 /src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp | |
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 'src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp')
-rw-r--r-- | src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp b/src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp index 6496a71251..0a0c0c50e8 100644 --- a/src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp +++ b/src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp @@ -36,8 +36,8 @@ FusedConvolutionBatchNormalizationNode::FusedConvolutionBatchNormalizationNode(f unsigned int num_groups, ConvolutionMethod method, FastMathHint fast_math_hint, - QuantizationInfo out_quant_info, ActivationLayerInfo fused_activation) - : _epsilon(epsilon), _info(std::move(info)), _num_groups(num_groups), _method(method), _fast_math_hint(fast_math_hint), _out_quant_info(std::move(out_quant_info)), _fused_activation(fused_activation) + ActivationLayerInfo fused_activation) + : _epsilon(epsilon), _info(std::move(info)), _num_groups(num_groups), _method(method), _fast_math_hint(fast_math_hint), _fused_activation(fused_activation) { _input_edges.resize(7, EmptyEdgeID); _outputs.resize(1, NullTensorID); @@ -132,10 +132,6 @@ TensorDescriptor FusedConvolutionBatchNormalizationNode::configure_output(size_t ARM_COMPUTE_ERROR_ON(src == nullptr || weights == nullptr); TensorDescriptor output_info = compute_output_descriptor(src->desc(), weights->desc(), _info); - if(!_out_quant_info.empty()) - { - output_info.quant_info = _out_quant_info; - } return output_info; } |