diff options
Diffstat (limited to 'arm_compute/graph/nodes/QuantizationLayerNode.h')
-rw-r--r-- | arm_compute/graph/nodes/QuantizationLayerNode.h | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/arm_compute/graph/nodes/QuantizationLayerNode.h b/arm_compute/graph/nodes/QuantizationLayerNode.h index f283c26996..b8e4c7d27b 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,16 +40,24 @@ 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; TensorDescriptor configure_output(size_t idx) const override; - void accept(INodeVisitor &v) override; + void accept(INodeVisitor &v) override; static constexpr NodeType node_type = NodeType::QuantizationLayer; private: QuantizationInfo _out_quant_info; + DataType _out_data_type; }; } // namespace graph } // namespace arm_compute |