src/armnn/InternalTypes.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // See LICENSE file in the project root for full license information.
   4 //
   5
   6 #include "InternalTypes.hpp"
   7
   8 #include <boost/assert.hpp>
   9
  10 namespace armnn
  11 {
  12
  13 char const* GetLayerTypeAsCString(LayerType type)
  14 {
  15     switch (type)
  16     {
  17         case LayerType::Activation: return "Activation";
  18         case LayerType::Addition: return "Addition";
  19         case LayerType::BatchNormalization: return "BatchNormalization";
  20         case LayerType::Constant: return "Constant";
  21         case LayerType::ConvertFp16ToFp32: return "ConvertFp16ToFp32";
  22         case LayerType::ConvertFp32ToFp16: return "ConvertFp32ToFp16";
  23         case LayerType::Convolution2d: return "Convolution2d";
  24         case LayerType::DepthwiseConvolution2d: return "DepthwiseConvolution2d";
  25         case LayerType::FakeQuantization: return "FakeQuantization";
  26         case LayerType::Floor: return "Floor";
  27         case LayerType::FullyConnected: return "FullyConnected";
  28         case LayerType::Input: return "Input";
  29         case LayerType::L2Normalization: return "L2Normalization";
  30         case LayerType::Lstm: return "Lstm";
  31         case LayerType::MemCopy: return "MemCopy";
  32         case LayerType::Merger: return "Merger";
  33         case LayerType::Multiplication: return "Multiplication";
  34         case LayerType::Normalization: return "Normalization";
  35         case LayerType::Output: return "Output";
  36         case LayerType::Permute: return "Permute";
  37         case LayerType::Pooling2d: return "Pooling2d";
  38         case LayerType::Reshape: return "Reshape";
  39         case LayerType::ResizeBilinear: return "ResizeBilinear";
  40         case LayerType::Softmax: return "Softmax";
  41         case LayerType::Splitter: return "Splitter";
  42         default:
  43             BOOST_ASSERT_MSG(false, "Unknown layer type");
  44             return "Unknown";
  45     }
  46 }
  47
  48 }