aboutsummaryrefslogtreecommitdiff
path: root/src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp
diff options
context:
space:
mode:
authorManuel Bottini <manuel.bottini@arm.com>2019-06-20 16:00:27 +0100
committerManuel Bottini <manuel.bottini@arm.com>2019-07-11 16:52:18 +0000
commitbffb41e06c1276af00e1605ef934d05fa61f7127 (patch)
tree7c9cfe90e82a8107ad8e32272c4e40c4b63182ef /src/graph/nodes/FusedConvolutionBatchNormalizationNode.cpp
parentc1b76faf6be5c33dbf3269faea95e185ac37992f (diff)
downloadComputeLibrary-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.cpp8
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;
}