diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-06-01 17:49:09 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:52:54 +0000 |
commit | da2491fb6d3cefb69846f220356fff282486495c (patch) | |
tree | 6d3106048be737ef65d7d61c936ba1aee54001a4 /arm_compute/graph/TypePrinter.h | |
parent | ec4c3201cba6ebf1c9d5ccad3bb26ad639f85cbb (diff) | |
download | ComputeLibrary-da2491fb6d3cefb69846f220356fff282486495c.tar.gz |
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>
Diffstat (limited to 'arm_compute/graph/TypePrinter.h')
-rw-r--r-- | arm_compute/graph/TypePrinter.h | 71 |
1 files changed, 71 insertions, 0 deletions
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) { |