diff options
Diffstat (limited to 'arm_compute/graph/nodes/DeconvolutionLayerNode.h')
-rw-r--r-- | arm_compute/graph/nodes/DeconvolutionLayerNode.h | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/arm_compute/graph/nodes/DeconvolutionLayerNode.h b/arm_compute/graph/nodes/DeconvolutionLayerNode.h index 5633898366..a5efdfb3bc 100644 --- a/arm_compute/graph/nodes/DeconvolutionLayerNode.h +++ b/arm_compute/graph/nodes/DeconvolutionLayerNode.h @@ -25,6 +25,7 @@ #define ARM_COMPUTE_GRAPH_DECONVOLUTION_LAYER_NODE_H #include "arm_compute/graph/INode.h" +#include "arm_compute/graph/LayerDescriptors.h" namespace arm_compute { @@ -36,10 +37,9 @@ class DeconvolutionLayerNode final : public INode public: /** Constructor * - * @param[in] info DeConvolution layer attributes - * @param[in] out_quant_info (Optional) Output quantization infomation + * @param[in] descriptor Contains information used by this layer described in @ref descriptors::DeconvolutionLayerDescriptor */ - DeconvolutionLayerNode(PadStrideInfo info, QuantizationInfo out_quant_info = QuantizationInfo()); + DeconvolutionLayerNode(const descriptors::DeconvolutionLayerDescriptor &descriptor); /** Deconvolution metadata accessor * * @return Deconvolution information @@ -64,8 +64,7 @@ public: void accept(INodeVisitor &v) override; private: - PadStrideInfo _info; - QuantizationInfo _out_quant_info; + descriptors::DeconvolutionLayerDescriptor descriptor; }; } // namespace graph } // namespace arm_compute |