From 0a05e6a051e6d9028131de10f00b80bbedf44097 Mon Sep 17 00:00:00 2001
From: thecha01 <theo.charalambous@arm.com>
Date: Fri, 28 Aug 2020 18:40:38 +0100
Subject: Add DepthToSpace layer node to Graph API

Change-Id: I5db780fb9c94160130e9986bbfc739124bfa8041
Signed-off-by: thecha01 <theo.charalambous@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3914
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
---
 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 5d6c9f308d..281de36bb2 100644
--- a/arm_compute/graph/TypePrinter.h
+++ b/arm_compute/graph/TypePrinter.h
@@ -83,6 +83,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type)
         case NodeType::DeconvolutionLayer:
             os << "DeconvolutionLayer";
             break;
+        case NodeType::DepthToSpaceLayer:
+            os << "DepthToSpaceLayer";
+            break;
         case NodeType::DequantizationLayer:
             os << "DequantizationLayer";
             break;
-- 
cgit v1.2.1