aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/LayerDescriptors.h
diff options
context:
space:
mode:
authorSang-Hoon Park <sang-hoon.park@arm.com>2020-03-13 11:31:53 +0000
committerSang-Hoon Park <sang-hoon.park@arm.com>2020-03-13 16:11:38 +0000
commit104fbd7b533c40f19465c85e884f10ae500e639e (patch)
tree10b6fb262fe72e5afeb13ba5b8f491fb3c1a825e /arm_compute/graph/LayerDescriptors.h
parent797b76b1aef38ea3be6f68ae2bf323048e9beff8 (diff)
downloadComputeLibrary-104fbd7b533c40f19465c85e884f10ae500e639e.tar.gz
COMPMID-3221: Add DeconvolutionLayerDescriptor
A new struct for DeconvolutionLayerNode is added for better extendability. Change-Id: I935277e8073a8295de7b0059b946cb637085f1ff Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2883 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/graph/LayerDescriptors.h')
-rw-r--r--arm_compute/graph/LayerDescriptors.h16
1 files changed, 16 insertions, 0 deletions
diff --git a/arm_compute/graph/LayerDescriptors.h b/arm_compute/graph/LayerDescriptors.h
index af69682fa3..0cf203174e 100644
--- a/arm_compute/graph/LayerDescriptors.h
+++ b/arm_compute/graph/LayerDescriptors.h
@@ -86,6 +86,22 @@ struct EltwiseLayerDescriptor
RoundingPolicy r_policy; /**< Rounding policy */
};
+/** Deconvolution layer descriptor */
+struct DeconvolutionLayerDescriptor
+{
+ /** Constructor
+ *
+ * @param[in] info Dedonvolution layer attributes
+ * @param[in] out_quant_info (Optional) Output quantization infomation
+ */
+ DeconvolutionLayerDescriptor(PadStrideInfo info, QuantizationInfo out_quant_info = QuantizationInfo())
+ : info(info), out_quant_info(out_quant_info)
+ {
+ }
+
+ PadStrideInfo info; /**< Padding and stride information */
+ QuantizationInfo out_quant_info; /**< Output quantization information */
+};
} // namespace descriptor
} // namespace graph
} // namespace arm_compute