Use X macro for the enum class LayerType

[platform/upstream/armnn.git] / src / armnn / InternalTypes.cpp

diff --git a/src/armnn/InternalTypes.cpp b/src/armnn/InternalTypes.cpp
index a9435b2..aebc721 100644 (file)
--- a/src/armnn/InternalTypes.cpp
+++ b/src/armnn/InternalTypes.cpp
@@ -14,66 +14,9 @@ char const* GetLayerTypeAsCString(LayerType type)
 {
     switch (type)
     {
-        case LayerType::Activation: return "Activation";
-        case LayerType::Addition: return "Addition";
-        case LayerType::ArgMinMax: return "ArgMinMax";
-        case LayerType::BatchNormalization: return "BatchNormalization";
-        case LayerType::BatchToSpaceNd: return "BatchToSpaceNd";
-        case LayerType::Comparison: return "Comparison";
-        case LayerType::Concat: return "Concat";
-        case LayerType::Constant: return "Constant";
-        case LayerType::ConvertBf16ToFp32: return "ConvertBf16ToFp32";
-        case LayerType::ConvertFp16ToFp32: return "ConvertFp16ToFp32";
-        case LayerType::ConvertFp32ToBf16: return "ConvertFp32ToBf16";
-        case LayerType::ConvertFp32ToFp16: return "ConvertFp32ToFp16";
-        case LayerType::Convolution2d: return "Convolution2d";
-        case LayerType::Debug: return "Debug";
-        case LayerType::DepthToSpace: return "DepthToSpace";
-        case LayerType::DepthwiseConvolution2d: return "DepthwiseConvolution2d";
-        case LayerType::Dequantize: return "Dequantize";
-        case LayerType::DetectionPostProcess: return "DetectionPostProcess";
-        case LayerType::Division: return "Division";
-        case LayerType::ElementwiseUnary: return "ElementwiseUnary";
-        case LayerType::FakeQuantization: return "FakeQuantization";
-        case LayerType::Floor: return "Floor";
-        case LayerType::FullyConnected: return "FullyConnected";
-        case LayerType::Gather: return "Gather";
-        case LayerType::Input: return "Input";
-        case LayerType::InstanceNormalization: return "InstanceNormalization";
-        case LayerType::L2Normalization: return "L2Normalization";
-        case LayerType::LogSoftmax: return "LogSoftmax";
-        case LayerType::Lstm: return "Lstm";
-        case LayerType::Maximum: return "Maximum";
-        case LayerType::Mean: return "Mean";
-        case LayerType::MemCopy: return "MemCopy";
-        case LayerType::MemImport: return "MemImport";
-        case LayerType::Merge: return "Merge";
-        case LayerType::Minimum: return "Minimum";
-        case LayerType::Multiplication: return "Multiplication";
-        case LayerType::Normalization: return "Normalization";
-        case LayerType::Output: return "Output";
-        case LayerType::Pad: return "Pad";
-        case LayerType::Permute: return "Permute";
-        case LayerType::Pooling2d: return "Pooling2d";
-        case LayerType::PreCompiled: return "PreCompiled";
-        case LayerType::Prelu: return "Prelu";
-        case LayerType::Quantize:  return "Quantize";
-        case LayerType::QLstm: return "QLstm";
-        case LayerType::QuantizedLstm: return "QuantizedLstm";
-        case LayerType::Reshape: return "Reshape";
-        case LayerType::Resize: return "Resize";
-        case LayerType::Slice: return "Slice";
-        case LayerType::Softmax: return "Softmax";
-        case LayerType::SpaceToBatchNd: return "SpaceToBatchNd";
-        case LayerType::SpaceToDepth: return "SpaceToDepth";
-        case LayerType::Splitter: return "Splitter";
-        case LayerType::Stack: return "Stack";
-        case LayerType::StandIn: return "StandIn";
-        case LayerType::StridedSlice: return "StridedSlice";
-        case LayerType::Subtraction: return "Subtraction";
-        case LayerType::Switch: return "Switch";
-        case LayerType::TransposeConvolution2d: return "TransposeConvolution2d";
-        case LayerType::Transpose: return "Transpose";
+#define X(name) case LayerType::name: return #name;
+      LIST_OF_LAYER_TYPE
+#undef X
         default:
             ARMNN_ASSERT_MSG(false, "Unknown layer type");
             return "Unknown";