ArmNN
 25.02
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Public Member Functions | List of all members
SerializerStrategy Class Reference

#include <Serializer.hpp>

Inheritance diagram for SerializerStrategy:
[legend]
Collaboration diagram for SerializerStrategy:
[legend]

Public Member Functions

void ExecuteStrategy (const armnn::IConnectableLayer *layer, const armnn::BaseDescriptor &descriptor, const std::vector< armnn::ConstTensor > &constants, const char *name, const armnn::LayerBindingId id) override
 
 SerializerStrategy ()
 
 ~SerializerStrategy ()
 
flatbuffers::FlatBufferBuilder & GetFlatBufferBuilder ()
 
std::vector< int > & GetInputIds ()
 
std::vector< int > & GetOutputIds ()
 
std::vector< flatbuffers::Offset< armnnSerializer::AnyLayer > > & GetSerializedLayers ()
 
flatbuffers::Offset< armnnSerializer::FeatureCompatibilityVersions > GetVersionTable ()
 
- Public Member Functions inherited from IStrategy
virtual void ExecuteStrategy (const IConnectableLayer *layer, const armnn::BaseDescriptor &descriptor, const std::vector< armnn::ConstTensor > &constants, const char *name, const armnn::LayerBindingId id=0)=0
 
virtual void FinishStrategy ()
 

Additional Inherited Members

- Protected Member Functions inherited from IStrategy
 IStrategy ()
 
virtual ~IStrategy ()
 

Detailed Description

Definition at line 22 of file Serializer.hpp.

Constructor & Destructor Documentation

◆ SerializerStrategy()

SerializerStrategy ( )
inline

Definition at line 31 of file Serializer.hpp.

31 : m_layerId(0) {}

◆ ~SerializerStrategy()

~SerializerStrategy ( )
inline

Definition at line 32 of file Serializer.hpp.

32 {}

Member Function Documentation

◆ ExecuteStrategy()

void ExecuteStrategy ( const armnn::IConnectableLayer *  layer,
const armnn::BaseDescriptor &  descriptor,
const std::vector< armnn::ConstTensor > &  constants,
const char *  name,
const armnn::LayerBindingId  id 
)
override

Definition at line 2074 of file Serializer.cpp.

2079 {
2080  IgnoreUnused(constants);
2081 
2082  switch (layer->GetType())
2083  {
2084  case armnn::LayerType::Activation :
2085  {
2086  const armnn::ActivationDescriptor& layerDescriptor =
2087  static_cast<const armnn::ActivationDescriptor&>(descriptor);
2088  SerializeActivationLayer(layer, layerDescriptor, name);
2089  break;
2090  }
2091  case armnn::LayerType::Addition :
2092  {
2093  SerializeAdditionLayer(layer, name);
2094  break;
2095  }
2096  case armnn::LayerType::ArgMinMax :
2097  {
2098  const armnn::ArgMinMaxDescriptor& layerDescriptor =
2099  static_cast<const armnn::ArgMinMaxDescriptor&>(descriptor);
2100  SerializeArgMinMaxLayer(layer, layerDescriptor, name);
2101  break;
2102  }
2103  case armnn::LayerType::BatchMatMul:
2104  {
2105  const armnn::BatchMatMulDescriptor& layerDescriptor =
2106  static_cast<const armnn::BatchMatMulDescriptor&>(descriptor);
2107  SerializeBatchMatMulLayer(layer,
2108  layerDescriptor,
2109  name);
2110  break;
2111  }
2112  case armnn::LayerType::BatchNormalization :
2113  {
2114  const armnn::BatchNormalizationDescriptor& layerDescriptor =
2115  static_cast<const armnn::BatchNormalizationDescriptor&>(descriptor);
2116  SerializeBatchNormalizationLayer(layer,
2117  layerDescriptor,
2118  constants,
2119  name);
2120  break;
2121  }
2122  case armnn::LayerType::BatchToSpaceNd :
2123  {
2124  const armnn::BatchToSpaceNdDescriptor& layerDescriptor =
2125  static_cast<const armnn::BatchToSpaceNdDescriptor&>(descriptor);
2126  SerializeBatchToSpaceNdLayer(layer,
2127  layerDescriptor,
2128  name);
2129  break;
2130  }
2131  case armnn::LayerType::Cast :
2132  {
2133  SerializeCastLayer(layer, name);
2134  break;
2135  }
2136  case armnn::LayerType::ChannelShuffle :
2137  {
2138  const armnn::ChannelShuffleDescriptor& layerDescriptor =
2139  static_cast<const armnn::ChannelShuffleDescriptor&>(descriptor);
2140  SerializeChannelShuffleLayer(layer,
2141  layerDescriptor,
2142  name);
2143  break;
2144  }
2145  case armnn::LayerType::Comparison :
2146  {
2147  const armnn::ComparisonDescriptor& layerDescriptor =
2148  static_cast<const armnn::ComparisonDescriptor&>(descriptor);
2149  SerializeComparisonLayer(layer,
2150  layerDescriptor,
2151  name);
2152  break;
2153  }
2154  case armnn::LayerType::Concat :
2155  {
2156  const armnn::ConcatDescriptor& layerDescriptor =
2157  static_cast<const armnn::ConcatDescriptor&>(descriptor);
2158  SerializeConcatLayer(layer,
2159  layerDescriptor,
2160  name);
2161  break;
2162  }
2163  case armnn::LayerType::Constant :
2164  {
2165  SerializeConstantLayer(layer,
2166  constants,
2167  name);
2168  break;
2169  }
2170  case armnn::LayerType::Convolution2d :
2171  {
2172  const armnn::Convolution2dDescriptor& layerDescriptor =
2173  static_cast<const armnn::Convolution2dDescriptor&>(descriptor);
2174  SerializeConvolution2dLayer(layer,
2175  layerDescriptor,
2176  name);
2177  break;
2178  }
2179  case armnn::LayerType::Convolution3d :
2180  {
2181  const armnn::Convolution3dDescriptor& layerDescriptor =
2182  static_cast<const armnn::Convolution3dDescriptor&>(descriptor);
2183  SerializeConvolution3dLayer(layer,
2184  layerDescriptor,
2185  name);
2186  break;
2187  }
2188  case armnn::LayerType::DepthToSpace :
2189  {
2190  const armnn::DepthToSpaceDescriptor& layerDescriptor =
2191  static_cast<const armnn::DepthToSpaceDescriptor&>(descriptor);
2192  SerializeDepthToSpaceLayer(layer,
2193  layerDescriptor,
2194  name);
2195  break;
2196  }
2197  case armnn::LayerType::DepthwiseConvolution2d :
2198  {
2199  const armnn::DepthwiseConvolution2dDescriptor& layerDescriptor =
2200  static_cast<const armnn::DepthwiseConvolution2dDescriptor&>(descriptor);
2201  SerializeDepthwiseConvolution2dLayer(layer,
2202  layerDescriptor,
2203  name);
2204  break;
2205  }
2206  case armnn::LayerType::Dequantize :
2207  {
2208  SerializeDequantizeLayer(layer,
2209  name);
2210  break;
2211  }
2212  case armnn::LayerType::DetectionPostProcess :
2213  {
2214  const armnn::DetectionPostProcessDescriptor& layerDescriptor =
2215  static_cast<const armnn::DetectionPostProcessDescriptor&>(descriptor);
2216  SerializeDetectionPostProcessLayer(layer, layerDescriptor, constants, name);
2217  break;
2218  }
2219  case armnn::LayerType::Division :
2220  {
2221  SerializeDivisionLayer(layer, name);
2222  break;
2223  }
2224  case armnn::LayerType::ElementwiseBinary :
2225  {
2226  const armnn::ElementwiseBinaryDescriptor& layerDescriptor =
2227  static_cast<const armnn::ElementwiseBinaryDescriptor&>(descriptor);
2228  SerializeElementwiseBinaryLayer(layer, layerDescriptor, name);
2229  break;
2230  }
2231  case armnn::LayerType::ElementwiseUnary :
2232  {
2233  const armnn::ElementwiseUnaryDescriptor& layerDescriptor =
2234  static_cast<const armnn::ElementwiseUnaryDescriptor&>(descriptor);
2235  SerializeElementwiseUnaryLayer(layer, layerDescriptor, name);
2236  break;
2237  }
2238  case armnn::LayerType::Fill :
2239  {
2240  const armnn::FillDescriptor& layerDescriptor =
2241  static_cast<const armnn::FillDescriptor&>(descriptor);
2242  SerializeFillLayer(layer, layerDescriptor, name);
2243  break;
2244  }
2245  case armnn::LayerType::Floor :
2246  {
2247  SerializeFloorLayer(layer, name);
2248  break;
2249  }
2250  case armnn::LayerType::FullyConnected :
2251  {
2252  const armnn::FullyConnectedDescriptor& layerDescriptor =
2253  static_cast<const armnn::FullyConnectedDescriptor&>(descriptor);
2254  SerializeFullyConnectedLayer(layer, layerDescriptor, name);
2255  break;
2256  }
2257  case armnn::LayerType::Gather :
2258  {
2259  const armnn::GatherDescriptor& layerDescriptor =
2260  static_cast<const armnn::GatherDescriptor&>(descriptor);
2261  SerializeGatherLayer(layer, layerDescriptor, name);
2262  break;
2263  }
2264  case armnn::LayerType::GatherNd :
2265  {
2266  SerializeGatherNdLayer(layer, name);
2267  break;
2268  }
2269  case armnn::LayerType::Input:
2270  {
2271  SerializeInputLayer(layer, id, name);
2272  break;
2273  }
2274  case armnn::LayerType::InstanceNormalization :
2275  {
2276  const armnn::InstanceNormalizationDescriptor& layerDescriptor =
2277  static_cast<const armnn::InstanceNormalizationDescriptor&>(descriptor);
2278  SerializeInstanceNormalizationLayer(layer, layerDescriptor, name);
2279  break;
2280  }
2281  case armnn::LayerType::L2Normalization :
2282  {
2283  const armnn::L2NormalizationDescriptor& layerDescriptor =
2284  static_cast<const armnn::L2NormalizationDescriptor&>(descriptor);
2285  SerializeL2NormalizationLayer(layer, layerDescriptor, name);
2286  break;
2287  }
2288  case armnn::LayerType::LogicalBinary :
2289  {
2290  const armnn::LogicalBinaryDescriptor& layerDescriptor =
2291  static_cast<const armnn::LogicalBinaryDescriptor&>(descriptor);
2292  SerializeLogicalBinaryLayer(layer, layerDescriptor, name);
2293  break;
2294  }
2295  case armnn::LayerType::LogSoftmax :
2296  {
2297  const armnn::LogSoftmaxDescriptor& layerDescriptor =
2298  static_cast<const armnn::LogSoftmaxDescriptor&>(descriptor);
2299  SerializeLogSoftmaxLayer(layer, layerDescriptor, name);
2300  break;
2301  }
2302  case armnn::LayerType::Lstm :
2303  {
2304  const armnn::LstmDescriptor& layerDescriptor =
2305  static_cast<const armnn::LstmDescriptor&>(descriptor);
2306  SerializeLstmLayer(layer, layerDescriptor, constants, name);
2307  break;
2308  }
2309  case armnn::LayerType::QLstm :
2310  {
2311  const armnn::QLstmDescriptor& layerDescriptor =
2312  static_cast<const armnn::QLstmDescriptor&>(descriptor);
2313  SerializeQLstmLayer(layer, layerDescriptor, constants, name);
2314  break;
2315  }
2316  case armnn::LayerType::Maximum :
2317  {
2318  SerializeMaximumLayer(layer, name);
2319  break;
2320  }
2321  case armnn::LayerType::Mean :
2322  {
2323  const armnn::MeanDescriptor& layerDescriptor =
2324  static_cast<const armnn::MeanDescriptor&>(descriptor);
2325  SerializeMeanLayer(layer, layerDescriptor, name);
2326  break;
2327  }
2328  case armnn::LayerType::Merge :
2329  {
2330  SerializeMergeLayer(layer, name);
2331  break;
2332  }
2333  case armnn::LayerType::Minimum :
2334  {
2335  SerializeMinimumLayer(layer, name);
2336  break;
2337  }
2338  case armnn::LayerType::Multiplication :
2339  {
2340  SerializeMultiplicationLayer(layer, name);
2341  break;
2342  }
2343  case armnn::LayerType::Normalization :
2344  {
2345  const armnn::NormalizationDescriptor& layerDescriptor =
2346  static_cast<const armnn::NormalizationDescriptor&>(descriptor);
2347  SerializeNormalizationLayer(layer, layerDescriptor, name);
2348  break;
2349  }
2350  case armnn::LayerType::Output:
2351  {
2352  SerializeOutputLayer(layer, id, name);
2353  break;
2354  }
2355  case armnn::LayerType::Pad :
2356  {
2357  const armnn::PadDescriptor& layerDescriptor =
2358  static_cast<const armnn::PadDescriptor&>(descriptor);
2359  SerializePadLayer(layer, layerDescriptor, name);
2360  break;
2361  }
2362  case armnn::LayerType::Permute :
2363  {
2364  const armnn::PermuteDescriptor& layerDescriptor =
2365  static_cast<const armnn::PermuteDescriptor&>(descriptor);
2366  SerializePermuteLayer(layer, layerDescriptor, name);
2367  break;
2368  }
2369  case armnn::LayerType::Pooling2d :
2370  {
2371  const armnn::Pooling2dDescriptor& layerDescriptor =
2372  static_cast<const armnn::Pooling2dDescriptor&>(descriptor);
2373  SerializePooling2dLayer(layer, layerDescriptor, name);
2374  break;
2375  }
2376  case armnn::LayerType::Pooling3d :
2377  {
2378  const armnn::Pooling3dDescriptor& layerDescriptor =
2379  static_cast<const armnn::Pooling3dDescriptor&>(descriptor);
2380  SerializePooling3dLayer(layer, layerDescriptor, name);
2381  break;
2382  }
2383  case armnn::LayerType::Prelu :
2384  {
2385  SerializePreluLayer(layer, name);
2386  break;
2387  }
2388  case armnn::LayerType::Quantize :
2389  {
2390  SerializeQuantizeLayer(layer, name);
2391  break;
2392  }
2393  case armnn::LayerType::QuantizedLstm:
2394  SerializeQuantizedLstmLayer(layer, constants, name);
2395  break;
2396  case armnn::LayerType::Reshape:
2397  {
2398  const armnn::ReshapeDescriptor &layerDescriptor =
2399  static_cast<const armnn::ReshapeDescriptor &>(descriptor);
2400  SerializeReshapeLayer(layer, layerDescriptor, name);
2401  break;
2402  }
2403  case armnn::LayerType::Rank:
2404  {
2405  SerializeRankLayer(layer, name);
2406  break;
2407  }
2408  case armnn::LayerType::Reduce:
2409  {
2410  const armnn::ReduceDescriptor& layerDescriptor =
2411  static_cast<const armnn::ReduceDescriptor&>(descriptor);
2412  SerializeReduceLayer(layer, layerDescriptor, name);
2413  break;
2414  }
2415  case armnn::LayerType::Resize:
2416  {
2417  const armnn::ResizeDescriptor& layerDescriptor =
2418  static_cast<const armnn::ResizeDescriptor&>(descriptor);
2419  SerializeResizeLayer(layer, layerDescriptor, name);
2420  break;
2421  }
2422  case armnn::LayerType::ReverseV2:
2423  {
2424  SerializeReverseV2Layer(layer, name);
2425  break;
2426  }
2427  case armnn::LayerType::ScatterNd:
2428  {
2429  const armnn::ScatterNdDescriptor& layerDescriptor =
2430  static_cast<const armnn::ScatterNdDescriptor&>(descriptor);
2431  SerializeScatterNdLayer(layer, layerDescriptor, name);
2432  break;
2433  }
2434  case armnn::LayerType::Shape:
2435  {
2436  SerializeShapeLayer(layer, name);
2437  break;
2438  }
2439  case armnn::LayerType::Slice:
2440  {
2441  const armnn::SliceDescriptor& layerDescriptor =
2442  static_cast<const armnn::SliceDescriptor&>(descriptor);
2443  SerializeSliceLayer(layer, layerDescriptor, name);
2444  break;
2445  }
2446  case armnn::LayerType::Softmax:
2447  {
2448  const armnn::SoftmaxDescriptor& layerDescriptor =
2449  static_cast<const armnn::SoftmaxDescriptor&>(descriptor);
2450  SerializeSoftmaxLayer(layer, layerDescriptor, name);
2451  break;
2452  }
2453  case armnn::LayerType::SpaceToBatchNd:
2454  {
2455  const armnn::SpaceToBatchNdDescriptor& layerDescriptor =
2456  static_cast<const armnn::SpaceToBatchNdDescriptor&>(descriptor);
2457  SerializeSpaceToBatchNdLayer(layer, layerDescriptor, name);
2458  break;
2459  }
2460  case armnn::LayerType::SpaceToDepth:
2461  {
2462  const armnn::SpaceToDepthDescriptor& layerDescriptor =
2463  static_cast<const armnn::SpaceToDepthDescriptor&>(descriptor);
2464  SerializeSpaceToDepthLayer(layer, layerDescriptor, name);
2465  break;
2466  }
2467  case armnn::LayerType::Splitter:
2468  {
2469  const armnn::SplitterDescriptor& layerDescriptor =
2470  static_cast<const armnn::SplitterDescriptor&>(descriptor);
2471  SerializeSplitterLayer(layer, layerDescriptor, name);
2472  break;
2473  }
2474  case armnn::LayerType::Stack:
2475  {
2476  const armnn::StackDescriptor& layerDescriptor =
2477  static_cast<const armnn::StackDescriptor&>(descriptor);
2478  SerializeStackLayer(layer, layerDescriptor, name);
2479  break;
2480  }
2481  case armnn::LayerType::StandIn:
2482  {
2483  const armnn::StandInDescriptor& layerDescriptor =
2484  static_cast<const armnn::StandInDescriptor&>(descriptor);
2485  SerializeStandInLayer(layer, layerDescriptor, name);
2486  break;
2487  }
2488  case armnn::LayerType::StridedSlice:
2489  {
2490  const armnn::StridedSliceDescriptor& layerDescriptor =
2491  static_cast<const armnn::StridedSliceDescriptor&>(descriptor);
2492  SerializeStridedSliceLayer(layer, layerDescriptor, name);
2493  break;
2494  }
2495  case armnn::LayerType::Subtraction:
2496  {
2497  SerializeSubtractionLayer(layer, name);
2498  break;
2499  }
2500  case armnn::LayerType::Switch:
2501  {
2502  SerializeSwitchLayer(layer, name);
2503  break;
2504  }
2505  case armnn::LayerType::Tile:
2506  {
2507  const armnn::TileDescriptor& layerDescriptor =
2508  static_cast<const armnn::TileDescriptor&>(descriptor);
2509  SerializeTileLayer(layer, layerDescriptor, name);
2510  break;
2511  }
2512  case armnn::LayerType::Transpose:
2513  {
2514  const armnn::TransposeDescriptor& layerDescriptor =
2515  static_cast<const armnn::TransposeDescriptor&>(descriptor);
2516  SerializeTransposeLayer(layer, layerDescriptor, name);
2517  break;
2518  }
2519  case armnn::LayerType::TransposeConvolution2d:
2520  {
2521  const armnn::TransposeConvolution2dDescriptor& layerDescriptor =
2522  static_cast<const armnn::TransposeConvolution2dDescriptor&>(descriptor);
2523  SerializeTransposeConvolution2dLayer(layer, layerDescriptor, constants, name);
2524  break;
2525  }
2526  case armnn::LayerType::UnidirectionalSequenceLstm :
2527  {
2528  const armnn::UnidirectionalSequenceLstmDescriptor& layerDescriptor =
2529  static_cast<const armnn::UnidirectionalSequenceLstmDescriptor&>(descriptor);
2530  SerializeUnidirectionalSequenceLstmLayer(layer, layerDescriptor, constants, name);
2531  break;
2532  }
2533  default:
2534  {
2535  throw InvalidArgumentException(
2536  fmt::format("A layer of unknown type was given to the serializer. Layer name: {}; Layer Id: {}",
2537  layer->GetName(),
2538  id));
2539  }
2540  }
2541 }
armnn::IConnectableLayer::GetType
virtual LayerType GetType() const =0
Returns the armnn::LayerType of this layer.
armnn::IConnectableLayer::GetName
virtual const char * GetName() const =0
Returns the name of the layer.
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
armnn::ActivationDescriptor
An ActivationDescriptor for the ActivationLayer.
Definition: Descriptors.hpp:37
armnn::ArgMinMaxDescriptor
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition: Descriptors.hpp:68
armnn::BatchMatMulDescriptor
A BatchMatMulDescriptor for the BatchMatMul operator.
Definition: Descriptors.hpp:1585
armnn::BatchNormalizationDescriptor
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
Definition: Descriptors.hpp:829
armnn::BatchToSpaceNdDescriptor
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
Definition: Descriptors.hpp:876
armnn::ChannelShuffleDescriptor
A ChannelShuffleDescriptor for the ChannelShuffle operator.
Definition: Descriptors.hpp:1563
armnn::ComparisonDescriptor
A ComparisonDescriptor for the ComparisonLayer.
Definition: Descriptors.hpp:90
armnn::Convolution2dDescriptor
A Convolution2dDescriptor for the Convolution2dLayer.
Definition: Descriptors.hpp:535
armnn::Convolution3dDescriptor
A Convolution3dDescriptor for the Convolution3dLayer.
Definition: Descriptors.hpp:589
armnn::DepthwiseConvolution2dDescriptor
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
Definition: Descriptors.hpp:660
armnn::DetectionPostProcessDescriptor
Definition: Descriptors.hpp:714
armnn::ElementwiseBinaryDescriptor
A ElementwiseBinaryDescriptor for the ElementwiseBinaryLayer.
Definition: Descriptors.hpp:110
armnn::ElementwiseUnaryDescriptor
A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer.
Definition: Descriptors.hpp:130
armnn::FillDescriptor
A FillDescriptor for the FillLayer.
Definition: Descriptors.hpp:926
armnn::FullyConnectedDescriptor
A FullyConnectedDescriptor for the FullyConnectedLayer.
Definition: Descriptors.hpp:508
armnn::GatherDescriptor
A GatherDescriptor for the GatherLayer.
Definition: Descriptors.hpp:966
armnn::InstanceNormalizationDescriptor
An InstanceNormalizationDescriptor for InstanceNormalizationLayer.
Definition: Descriptors.hpp:848
armnn::L2NormalizationDescriptor
A L2NormalizationDescriptor for the L2NormalizationLayer.
Definition: Descriptors.hpp:810
armnn::LogicalBinaryDescriptor
A LogicalBinaryDescriptor for the LogicalBinaryLayer.
Definition: Descriptors.hpp:1519
armnn::LstmDescriptor
An LstmDescriptor for the LstmLayer.
Definition: Descriptors.hpp:1103
armnn::MeanDescriptor
A MeanDescriptor for the MeanLayer.
Definition: Descriptors.hpp:1173
armnn::NormalizationDescriptor
A NormalizationDescriptor for the NormalizationLayer.
Definition: Descriptors.hpp:770
armnn::OriginsDescriptor
An OriginsDescriptor for the ConcatLayer.
Definition: Descriptors.hpp:202
armnn::PadDescriptor
A PadDescriptor for the PadLayer.
Definition: Descriptors.hpp:1197
armnn::PermuteDescriptor
A PermuteDescriptor for the PermuteLayer.
Definition: Descriptors.hpp:150
armnn::Pooling2dDescriptor
A Pooling2dDescriptor for the Pooling2dLayer.
Definition: Descriptors.hpp:372
armnn::Pooling3dDescriptor
A Pooling3dDescriptor for the Pooling3dLayer.
Definition: Descriptors.hpp:432
armnn::QLstmDescriptor
A QLstmDescriptor for the QLstmLayer.
Definition: Descriptors.hpp:1381
armnn::ReduceDescriptor
A ReduceDescriptor for the REDUCE operators.
Definition: Descriptors.hpp:1539
armnn::ReshapeDescriptor
A ReshapeDescriptor for the ReshapeLayer.
Definition: Descriptors.hpp:1024
armnn::ResizeDescriptor
A ResizeDescriptor for the ResizeLayer.
Definition: Descriptors.hpp:986
armnn::ScatterNdDescriptor
A ScatterNdDescriptor for the ScatterNdLayer.
Definition: Descriptors.hpp:1680
armnn::SliceDescriptor
A SliceDescriptor for the SliceLayer.
Definition: Descriptors.hpp:1229
armnn::SoftmaxDescriptor
A SoftmaxDescriptor for the SoftmaxLayer.
Definition: Descriptors.hpp:178
armnn::SpaceToBatchNdDescriptor
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
Definition: Descriptors.hpp:1044
armnn::SpaceToDepthDescriptor
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
Definition: Descriptors.hpp:1076
armnn::StackDescriptor
A StackDescriptor for the StackLayer.
Definition: Descriptors.hpp:1252
armnn::StandInDescriptor
A StandInDescriptor for the StandIn layer.
Definition: Descriptors.hpp:1282
armnn::StridedSliceDescriptor
A StridedSliceDescriptor for the StridedSliceLayer.
Definition: Descriptors.hpp:1304
armnn::TransposeConvolution2dDescriptor
A TransposeConvolution2dDescriptor for the TransposeConvolution2dLayer.
Definition: Descriptors.hpp:1441
armnn::TransposeDescriptor
A TransposeDescriptor for the TransposeLayer.
Definition: Descriptors.hpp:1491
armnn::ViewsDescriptor
A ViewsDescriptor for the SplitterLayer.
Definition: Descriptors.hpp:245

References armnn::Activation, armnn::Addition, armnn::ArgMinMax, armnn::BatchMatMul, armnn::BatchNormalization, armnn::BatchToSpaceNd, armnn::Cast, armnn::ChannelShuffle, armnn::Comparison, armnn::Concat, armnn::Constant, armnn::Convolution2d, armnn::Convolution3d, armnn::DepthToSpace, armnn::DepthwiseConvolution2d, armnn::Dequantize, armnn::DetectionPostProcess, armnn::Division, armnn::ElementwiseBinary, armnn::ElementwiseUnary, armnn::Fill, armnn::Floor, armnn::FullyConnected, armnn::Gather, armnn::GatherNd, IConnectableLayer::GetName(), IConnectableLayer::GetType(), armnn::IgnoreUnused(), armnn::Input, armnn::InstanceNormalization, armnn::L2Normalization, armnn::LogicalBinary, armnn::LogSoftmax, armnn::Lstm, armnn::Maximum, armnn::Mean, armnn::Merge, armnn::Minimum, armnn::Multiplication, armnn::Normalization, armnn::Output, armnn::Pad, armnn::Permute, armnn::Pooling2d, armnn::Pooling3d, armnn::Prelu, armnn::QLstm, armnn::Quantize, armnn::QuantizedLstm, armnn::Rank, armnn::Reduce, armnn::Reshape, armnn::Resize, armnn::ReverseV2, armnn::ScatterNd, armnn::Shape, armnn::Slice, armnn::Softmax, armnn::SpaceToBatchNd, armnn::SpaceToDepth, armnn::Splitter, armnn::Stack, armnn::StandIn, armnn::StridedSlice, armnn::Subtraction, armnn::Switch, armnn::Tile, armnn::Transpose, armnn::TransposeConvolution2d, and armnn::UnidirectionalSequenceLstm.

◆ GetFlatBufferBuilder()

flatbuffers::FlatBufferBuilder& GetFlatBufferBuilder ( )
inline

Definition at line 34 of file Serializer.hpp.

35  {
36  return m_flatBufferBuilder;
37  }

◆ GetInputIds()

std::vector<int>& GetInputIds ( )
inline

Definition at line 39 of file Serializer.hpp.

40  {
41  return m_inputIds;
42  }

◆ GetOutputIds()

std::vector<int>& GetOutputIds ( )
inline

Definition at line 44 of file Serializer.hpp.

45  {
46  return m_outputIds;
47  }

◆ GetSerializedLayers()

std::vector<flatbuffers::Offset<armnnSerializer::AnyLayer> >& GetSerializedLayers ( )
inline

Definition at line 49 of file Serializer.hpp.

50  {
51  return m_serializedLayers;
52  }

◆ GetVersionTable()

flatbuffers::Offset< armnnSerializer::FeatureCompatibilityVersions > GetVersionTable ( )

Definition at line 2017 of file Serializer.cpp.

2018 {
2019  flatbuffers::Offset<armnnSerializer::FeatureCompatibilityVersions> versionsTable =
2020  serializer::CreateFeatureCompatibilityVersions(
2021  m_flatBufferBuilder,
2022  1, // Binding ids scheme version
2023  1, // Weights layout scheme version
2024  1 // Constant tensors as inputs version
2025  );
2026  return versionsTable;
2027 }
The documentation for this class was generated from the following files: