From 96a14008af85725d067cdd8247023474581102ea Mon Sep 17 00:00:00 2001 From: thecha01 Date: Tue, 28 Jul 2020 17:45:07 +0100 Subject: Fix EltwiseLayerNode and QuantizationLayerNode - Fixed issue where EltwiseLayerNode would base output shape off of first input tensor only - Allow QuantizationLayerNode to use any quantized data type if specified in constructor Signed-off-by: thecha01 Change-Id: Ib93470316799028cd573592a3d79943493eaa093 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3737 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Manuel Bottini --- arm_compute/graph/nodes/QuantizationLayerNode.h | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'arm_compute/graph') diff --git a/arm_compute/graph/nodes/QuantizationLayerNode.h b/arm_compute/graph/nodes/QuantizationLayerNode.h index 94c718babb..e5d81afa0e 100644 --- a/arm_compute/graph/nodes/QuantizationLayerNode.h +++ b/arm_compute/graph/nodes/QuantizationLayerNode.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019 Arm Limited. + * Copyright (c) 2019-2020 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -40,6 +40,13 @@ public: */ QuantizationLayerNode(QuantizationInfo out_quant_info); + /** Constructor + * + * @param[in] out_quant_info Output quantization info + * @param[in] out_data_type Output data type + */ + QuantizationLayerNode(QuantizationInfo out_quant_info, DataType out_data_type); + // Inherited overridden methods: NodeType type() const override; bool forward_descriptors() override; @@ -50,6 +57,7 @@ public: private: QuantizationInfo _out_quant_info; + DataType _out_data_type; }; } // namespace graph } // namespace arm_compute -- cgit v1.2.1