diff options
Diffstat (limited to 'src/graph/nodes/ActivationLayerNode.cpp')
-rw-r--r-- | src/graph/nodes/ActivationLayerNode.cpp | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/src/graph/nodes/ActivationLayerNode.cpp b/src/graph/nodes/ActivationLayerNode.cpp index 85cb10bbdb..ada6cf981f 100644 --- a/src/graph/nodes/ActivationLayerNode.cpp +++ b/src/graph/nodes/ActivationLayerNode.cpp @@ -30,8 +30,8 @@ namespace arm_compute { namespace graph { -ActivationLayerNode::ActivationLayerNode(ActivationLayerInfo info) - : _info(info) +ActivationLayerNode::ActivationLayerNode(ActivationLayerInfo info, QuantizationInfo out_quant_info) + : _info(info), _out_quant_info(out_quant_info) { _input_edges.resize(1, EmptyEdgeID); _outputs.resize(1, NullTensorID); @@ -62,7 +62,13 @@ TensorDescriptor ActivationLayerNode::configure_output(size_t idx) const const Tensor *src = input(0); ARM_COMPUTE_ERROR_ON(src == nullptr); - return src->desc(); + TensorDescriptor output_info = src->desc(); + if(!_out_quant_info.empty()) + { + output_info.quant_info = _out_quant_info; + } + + return output_info; } NodeType ActivationLayerNode::type() const |