aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/nodes/DeconvolutionLayerNode.h
diff options
context:
space:
mode:
authorSang-Hoon Park <sang-hoon.park@arm.com>2020-03-06 16:32:01 +0000
committerGeorgios Pinitas <georgios.pinitas@arm.com>2020-03-11 18:34:03 +0000
commit6800117df3be825f0ec5c6cc71c4377322f51b99 (patch)
tree0e579a271f2676dc2d6aa947df29a5cf0ab8bd1c /arm_compute/graph/nodes/DeconvolutionLayerNode.h
parent9204646e091ffc25eda61768537357916a4f7df4 (diff)
downloadComputeLibrary-6800117df3be825f0ec5c6cc71c4377322f51b99.tar.gz
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>
Diffstat (limited to 'arm_compute/graph/nodes/DeconvolutionLayerNode.h')
-rw-r--r--arm_compute/graph/nodes/DeconvolutionLayerNode.h10
1 files changed, 6 insertions, 4 deletions
diff --git a/arm_compute/graph/nodes/DeconvolutionLayerNode.h b/arm_compute/graph/nodes/DeconvolutionLayerNode.h
index 09601ab69e..5633898366 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 @@ class DeconvolutionLayerNode final : public INode
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 @@ public:
void accept(INodeVisitor &v) override;
private:
- PadStrideInfo _info;
+ PadStrideInfo _info;
+ QuantizationInfo _out_quant_info;
};
} // namespace graph
} // namespace arm_compute