From d64444ba197c2f95dcf4d205f50a196d5a29cdeb Mon Sep 17 00:00:00 2001 From: thecha01 Date: Mon, 7 Sep 2020 14:50:21 +0100 Subject: Add Reduction operation layer node to graph API Signed-off-by: thecha01 Change-Id: Ida819fb8c33790cc9da6d69eeb51e0599269197a Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3931 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- arm_compute/graph/TypePrinter.h | 3 +++ 1 file changed, 3 insertions(+) (limited to 'arm_compute/graph/TypePrinter.h') diff --git a/arm_compute/graph/TypePrinter.h b/arm_compute/graph/TypePrinter.h index 738bffaf15..22b96a04c3 100644 --- a/arm_compute/graph/TypePrinter.h +++ b/arm_compute/graph/TypePrinter.h @@ -151,6 +151,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) case NodeType::QuantizationLayer: os << "QuantizationLayer"; break; + case NodeType::ReductionOperationLayer: + os << "ReductionOperationLayer"; + break; case NodeType::ReorgLayer: os << "ReorgLayer"; break; -- cgit v1.2.1