diff options
Diffstat (limited to 'arm_compute/graph/nodes/ActivationLayerNode.h')
-rw-r--r-- | arm_compute/graph/nodes/ActivationLayerNode.h | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/arm_compute/graph/nodes/ActivationLayerNode.h b/arm_compute/graph/nodes/ActivationLayerNode.h index 723120655b..a17b0103e4 100644 --- a/arm_compute/graph/nodes/ActivationLayerNode.h +++ b/arm_compute/graph/nodes/ActivationLayerNode.h @@ -36,9 +36,11 @@ class ActivationLayerNode final : public INode public: /** Constructor * - * @param[in] info Activation Layer information + * @param[in] info Activation Layer information + * @param[in] out_quant_info (Optional) Output quantization info */ - ActivationLayerNode(ActivationLayerInfo info); + ActivationLayerNode(ActivationLayerInfo info, + QuantizationInfo out_quant_info = QuantizationInfo()); /** Activation metadata accessor * * @return The activation info of the layer @@ -56,6 +58,7 @@ public: private: ActivationLayerInfo _info; + QuantizationInfo _out_quant_info; }; } // namespace graph } // namespace arm_compute |