COMPMID-3221: (DATA_UPDATE) add graph example for EDSR
Change-Id: Id74190e2af444da8dab4813fd65016104f3882a9
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2862
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
diff --git a/arm_compute/graph/nodes/DeconvolutionLayerNode.h b/arm_compute/graph/nodes/DeconvolutionLayerNode.h
index 09601ab..5633898 100644
--- a/arm_compute/graph/nodes/DeconvolutionLayerNode.h
+++ b/arm_compute/graph/nodes/DeconvolutionLayerNode.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018-2019 ARM Limited.
+ * Copyright (c) 2018-2020 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -36,9 +36,10 @@
public:
/** Constructor
*
- * @param[in] info DeConvolution layer attributes
+ * @param[in] info DeConvolution layer attributes
+ * @param[in] out_quant_info (Optional) Output quantization infomation
*/
- DeconvolutionLayerNode(PadStrideInfo info);
+ DeconvolutionLayerNode(PadStrideInfo info, QuantizationInfo out_quant_info = QuantizationInfo());
/** Deconvolution metadata accessor
*
* @return Deconvolution information
@@ -63,7 +64,8 @@
void accept(INodeVisitor &v) override;
private:
- PadStrideInfo _info;
+ PadStrideInfo _info;
+ QuantizationInfo _out_quant_info;
};
} // namespace graph
} // namespace arm_compute
diff --git a/arm_compute/graph/nodes/EltwiseLayerNode.h b/arm_compute/graph/nodes/EltwiseLayerNode.h
index 7fb0aff..1ca8e9c 100644
--- a/arm_compute/graph/nodes/EltwiseLayerNode.h
+++ b/arm_compute/graph/nodes/EltwiseLayerNode.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018-2019 ARM Limited.
+ * Copyright (c) 2018-2020 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -36,11 +36,12 @@
public:
/** Constructor
*
- * @param[in] op Element-wise operation to perform
- * @param[in] c_policy (Optional) Convert policy used for the operation
- * @param[in] r_policy (Optional) Rounding policy used for the operation
+ * @param[in] op Element-wise operation to perform
+ * @param[in] out_quant_info (Optional) Output quantization information
+ * @param[in] c_policy (Optional) Convert policy used for the operation
+ * @param[in] r_policy (Optional) Rounding policy used for the operation
*/
- EltwiseLayerNode(EltwiseOperation op, ConvertPolicy c_policy = ConvertPolicy::SATURATE, RoundingPolicy r_policy = RoundingPolicy::TO_ZERO);
+ EltwiseLayerNode(EltwiseOperation op, QuantizationInfo out_quant_info = QuantizationInfo(), ConvertPolicy c_policy = ConvertPolicy::SATURATE, RoundingPolicy r_policy = RoundingPolicy::TO_ZERO);
/** Eltwise operation accessor
*
* @return Eltwise operation that is to be performed by the node
@@ -67,6 +68,7 @@
private:
EltwiseOperation _op;
+ QuantizationInfo _out_quant_info;
ConvertPolicy _convert_policy;
RoundingPolicy _rounding_policy;
};