aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/TypePrinter.h
diff options
context:
space:
mode:
authorthecha01 <theo.charalambous@arm.com>2020-08-28 18:40:38 +0100
committerMichele Di Giorgio <michele.digiorgio@arm.com>2020-09-04 15:54:46 +0000
commit0a05e6a051e6d9028131de10f00b80bbedf44097 (patch)
tree4821a3efd2479e0a770442a30aad10e7db6b1fa0 /arm_compute/graph/TypePrinter.h
parentbe058f4b5fb2383e0fa4666a92c43ca994486c86 (diff)
downloadComputeLibrary-0a05e6a051e6d9028131de10f00b80bbedf44097.tar.gz
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>
Diffstat (limited to 'arm_compute/graph/TypePrinter.h')
-rw-r--r--arm_compute/graph/TypePrinter.h3
1 files changed, 3 insertions, 0 deletions
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;