aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/TypePrinter.h
diff options
context:
space:
mode:
authorIsabella Gottardi <isabella.gottardi@arm.com>2019-05-17 12:35:20 +0100
committerIsabella Gottardi <isabella.gottardi@arm.com>2019-05-20 07:51:02 +0000
commit3db1ba9938986e7096184ad22020a71775075d03 (patch)
tree5018e5eecc1c1c81277e8a4c37bc0ad76245d95e /arm_compute/graph/TypePrinter.h
parent0eb16cfa5ff4484658ba8630a8dea072445dacbd (diff)
downloadComputeLibrary-3db1ba9938986e7096184ad22020a71775075d03.tar.gz
COMPMID-2362: Add Quantization Layer to the graph API
Change-Id: I9766ad54e8da1367ba192dfeeccdc879f3729550 Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com> Reviewed-on: https://review.mlplatform.org/c/1172 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Tested-by: 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 29a2981463..4fb5b73333 100644
--- a/arm_compute/graph/TypePrinter.h
+++ b/arm_compute/graph/TypePrinter.h
@@ -122,6 +122,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type)
case NodeType::PriorBoxLayer:
os << "PriorBoxLayer";
break;
+ case NodeType::QuantizationLayer:
+ os << "QuantizationLayer";
+ break;
case NodeType::ReorgLayer:
os << "ReorgLayer";
break;