From da2491fb6d3cefb69846f220356fff282486495c Mon Sep 17 00:00:00 2001
From: Georgios Pinitas <georgios.pinitas@arm.com>
Date: Fri, 1 Jun 2018 17:49:09 +0100
Subject: COMPMID-1151: Templatize FunctionFactories.

Change-Id: Id1c68c3bf442c3fcff265041b260d007db7593cb
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/134027
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
---
 arm_compute/graph/TypePrinter.h | 71 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)

(limited to 'arm_compute/graph/TypePrinter.h')

diff --git a/arm_compute/graph/TypePrinter.h b/arm_compute/graph/TypePrinter.h
index edce77c4ad..36d924b4f9 100644
--- a/arm_compute/graph/TypePrinter.h
+++ b/arm_compute/graph/TypePrinter.h
@@ -156,6 +156,77 @@ inline ::std::ostream &operator<<(::std::ostream &os, const DataLayout &data_lay
     return os;
 }
 
+inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type)
+{
+    switch(node_type)
+    {
+        case NodeType::ActivationLayer:
+            os << "ActivationLayer";
+            break;
+        case NodeType::BatchNormalizationLayer:
+            os << "BatchNormalizationLayer";
+            break;
+        case NodeType::ChannelShuffleLayer:
+            os << "ChannelShuffleLayer";
+            break;
+        case NodeType::ConvolutionLayer:
+            os << "ConvolutionLayer";
+            break;
+        case NodeType::DeconvolutionLayer:
+            os << "DeconvolutionLayer";
+            break;
+        case NodeType::DepthConcatenateLayer:
+            os << "DepthConcatenateLayer";
+            break;
+        case NodeType::DepthwiseConvolutionLayer:
+            os << "DepthwiseConvolutionLayer";
+            break;
+        case NodeType::EltwiseLayer:
+            os << "EltwiseLayer";
+            break;
+        case NodeType::FlattenLayer:
+            os << "FlattenLayer";
+            break;
+        case NodeType::FullyConnectedLayer:
+            os << "FullyConnectedLayer";
+            break;
+        case NodeType::NormalizationLayer:
+            os << "NormalizationLayer";
+            break;
+        case NodeType::PoolingLayer:
+            os << "PoolingLayer";
+            break;
+        case NodeType::ReshapeLayer:
+            os << "ReshapeLayer";
+            break;
+        case NodeType::ResizeLayer:
+            os << "ResizeLayer";
+            break;
+        case NodeType::SoftmaxLayer:
+            os << "SoftmaxLayer";
+            break;
+        case NodeType::SplitLayer:
+            os << "SplitLayer";
+            break;
+        case NodeType::Input:
+            os << "Input";
+            break;
+        case NodeType::Output:
+            os << "Output";
+            break;
+        case NodeType::Const:
+            os << "Const";
+            break;
+        case NodeType::Dummy:
+            os << "Dummy";
+            break;
+        default:
+            ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
+    }
+
+    return os;
+}
+
 /** Formatted output of the activation function type. */
 inline ::std::ostream &operator<<(::std::ostream &os, const ActivationLayerInfo::ActivationFunction &act_function)
 {
-- 
cgit v1.2.1