ArmNN
 22.02
Public Member Functions | List of all members
SerializerStrategy Class Reference

#include <Serializer.hpp>

Inheritance diagram for SerializerStrategy:
IStrategy

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 armnn::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 24 of file Serializer.hpp.

Constructor & Destructor Documentation

◆ SerializerStrategy()

SerializerStrategy ( )
inline

Definition at line 33 of file Serializer.hpp.

33 : m_layerId(0) {}

◆ ~SerializerStrategy()

~SerializerStrategy ( )
inline

Definition at line 34 of file Serializer.hpp.

34 {}

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 1961 of file Serializer.cpp.

References armnn::Activation, armnn::Addition, armnn::ArgMinMax, 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::ElementwiseUnary, armnn::Fill, armnn::Floor, armnn::FullyConnected, armnn::Gather, 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::Shape, armnn::Slice, armnn::Softmax, armnn::SpaceToBatchNd, armnn::SpaceToDepth, armnn::Splitter, armnn::Stack, armnn::StandIn, armnn::StridedSlice, armnn::Subtraction, armnn::Switch, armnn::Transpose, armnn::TransposeConvolution2d, and armnn::UnidirectionalSequenceLstm.

1966 {
1967  IgnoreUnused(constants);
1968 
1969  switch (layer->GetType())
1970  {
1971  case armnn::LayerType::Activation :
1972  {
1973  const armnn::ActivationDescriptor& layerDescriptor =
1974  static_cast<const armnn::ActivationDescriptor&>(descriptor);
1975  SerializeActivationLayer(layer, layerDescriptor, name);
1976  break;
1977  }
1978  case armnn::LayerType::Addition :
1979  {
1980  SerializeAdditionLayer(layer, name);
1981  break;
1982  }
1983  case armnn::LayerType::ArgMinMax :
1984  {
1985  const armnn::ArgMinMaxDescriptor& layerDescriptor =
1986  static_cast<const armnn::ArgMinMaxDescriptor&>(descriptor);
1987  SerializeArgMinMaxLayer(layer, layerDescriptor, name);
1988  break;
1989  }
1990  case armnn::LayerType::BatchNormalization :
1991  {
1992  const armnn::BatchNormalizationDescriptor& layerDescriptor =
1993  static_cast<const armnn::BatchNormalizationDescriptor&>(descriptor);
1994  SerializeBatchNormalizationLayer(layer,
1995  layerDescriptor,
1996  constants,
1997  name);
1998  break;
1999  }
2000  case armnn::LayerType::BatchToSpaceNd :
2001  {
2002  const armnn::BatchToSpaceNdDescriptor& layerDescriptor =
2003  static_cast<const armnn::BatchToSpaceNdDescriptor&>(descriptor);
2004  SerializeBatchToSpaceNdLayer(layer,
2005  layerDescriptor,
2006  name);
2007  break;
2008  }
2009  case armnn::LayerType::Cast :
2010  {
2011  SerializeCastLayer(layer, name);
2012  break;
2013  }
2014  case armnn::LayerType::ChannelShuffle :
2015  {
2016  const armnn::ChannelShuffleDescriptor& layerDescriptor =
2017  static_cast<const armnn::ChannelShuffleDescriptor&>(descriptor);
2018  SerializeChannelShuffleLayer(layer,
2019  layerDescriptor,
2020  name);
2021  break;
2022  }
2023  case armnn::LayerType::Comparison :
2024  {
2025  const armnn::ComparisonDescriptor& layerDescriptor =
2026  static_cast<const armnn::ComparisonDescriptor&>(descriptor);
2027  SerializeComparisonLayer(layer,
2028  layerDescriptor,
2029  name);
2030  break;
2031  }
2032  case armnn::LayerType::Concat :
2033  {
2034  const armnn::ConcatDescriptor& layerDescriptor =
2035  static_cast<const armnn::ConcatDescriptor&>(descriptor);
2036  SerializeConcatLayer(layer,
2037  layerDescriptor,
2038  name);
2039  break;
2040  }
2041  case armnn::LayerType::Constant :
2042  {
2043  SerializeConstantLayer(layer,
2044  constants,
2045  name);
2046  break;
2047  }
2048  case armnn::LayerType::Convolution2d :
2049  {
2050  const armnn::Convolution2dDescriptor& layerDescriptor =
2051  static_cast<const armnn::Convolution2dDescriptor&>(descriptor);
2052  SerializeConvolution2dLayer(layer,
2053  layerDescriptor,
2054  constants,
2055  name);
2056  break;
2057  }
2058  case armnn::LayerType::Convolution3d :
2059  {
2060  const armnn::Convolution3dDescriptor& layerDescriptor =
2061  static_cast<const armnn::Convolution3dDescriptor&>(descriptor);
2062  SerializeConvolution3dLayer(layer,
2063  layerDescriptor,
2064  name);
2065  break;
2066  }
2067  case armnn::LayerType::DepthToSpace :
2068  {
2069  const armnn::DepthToSpaceDescriptor& layerDescriptor =
2070  static_cast<const armnn::DepthToSpaceDescriptor&>(descriptor);
2071  SerializeDepthToSpaceLayer(layer,
2072  layerDescriptor,
2073  name);
2074  break;
2075  }
2076  case armnn::LayerType::DepthwiseConvolution2d :
2077  {
2078  const armnn::DepthwiseConvolution2dDescriptor& layerDescriptor =
2079  static_cast<const armnn::DepthwiseConvolution2dDescriptor&>(descriptor);
2080  SerializeDepthwiseConvolution2dLayer(layer,
2081  layerDescriptor,
2082  constants,
2083  name);
2084  break;
2085  }
2086  case armnn::LayerType::Dequantize :
2087  {
2088  SerializeDequantizeLayer(layer,
2089  name);
2090  break;
2091  }
2092  case armnn::LayerType::DetectionPostProcess :
2093  {
2094  const armnn::DetectionPostProcessDescriptor& layerDescriptor =
2095  static_cast<const armnn::DetectionPostProcessDescriptor&>(descriptor);
2096  SerializeDetectionPostProcessLayer(layer, layerDescriptor, constants, name);
2097  break;
2098  }
2099  case armnn::LayerType::Division :
2100  {
2101  SerializeDivisionLayer(layer, name);
2102  break;
2103  }
2104  case armnn::LayerType::ElementwiseUnary :
2105  {
2106  const armnn::ElementwiseUnaryDescriptor& layerDescriptor =
2107  static_cast<const armnn::ElementwiseUnaryDescriptor&>(descriptor);
2108  SerializeElementwiseUnaryLayer(layer, layerDescriptor, name);
2109  break;
2110  }
2111  case armnn::LayerType::Fill :
2112  {
2113  const armnn::FillDescriptor& layerDescriptor =
2114  static_cast<const armnn::FillDescriptor&>(descriptor);
2115  SerializeFillLayer(layer, layerDescriptor, name);
2116  break;
2117  }
2118  case armnn::LayerType::Floor :
2119  {
2120  SerializeFloorLayer(layer, name);
2121  break;
2122  }
2123  case armnn::LayerType::FullyConnected :
2124  {
2125  const armnn::FullyConnectedDescriptor& layerDescriptor =
2126  static_cast<const armnn::FullyConnectedDescriptor&>(descriptor);
2127  SerializeFullyConnectedLayer(layer, layerDescriptor, name);
2128  break;
2129  }
2130  case armnn::LayerType::Gather :
2131  {
2132  const armnn::GatherDescriptor& layerDescriptor =
2133  static_cast<const armnn::GatherDescriptor&>(descriptor);
2134  SerializeGatherLayer(layer, layerDescriptor, name);
2135  break;
2136  }
2137  case armnn::LayerType::Input:
2138  {
2139  SerializeInputLayer(layer, id, name);
2140  break;
2141  }
2142  case armnn::LayerType::InstanceNormalization :
2143  {
2144  const armnn::InstanceNormalizationDescriptor& layerDescriptor =
2145  static_cast<const armnn::InstanceNormalizationDescriptor&>(descriptor);
2146  SerializeInstanceNormalizationLayer(layer, layerDescriptor, name);
2147  break;
2148  }
2149  case armnn::LayerType::L2Normalization :
2150  {
2151  const armnn::L2NormalizationDescriptor& layerDescriptor =
2152  static_cast<const armnn::L2NormalizationDescriptor&>(descriptor);
2153  SerializeL2NormalizationLayer(layer, layerDescriptor, name);
2154  break;
2155  }
2156  case armnn::LayerType::LogicalBinary :
2157  {
2158  const armnn::LogicalBinaryDescriptor& layerDescriptor =
2159  static_cast<const armnn::LogicalBinaryDescriptor&>(descriptor);
2160  SerializeLogicalBinaryLayer(layer, layerDescriptor, name);
2161  break;
2162  }
2163  case armnn::LayerType::LogSoftmax :
2164  {
2165  const armnn::LogSoftmaxDescriptor& layerDescriptor =
2166  static_cast<const armnn::LogSoftmaxDescriptor&>(descriptor);
2167  SerializeLogSoftmaxLayer(layer, layerDescriptor, name);
2168  break;
2169  }
2170  case armnn::LayerType::Lstm :
2171  {
2172  const armnn::LstmDescriptor& layerDescriptor =
2173  static_cast<const armnn::LstmDescriptor&>(descriptor);
2174  SerializeLstmLayer(layer, layerDescriptor, constants, name);
2175  break;
2176  }
2177  case armnn::LayerType::QLstm :
2178  {
2179  const armnn::QLstmDescriptor& layerDescriptor =
2180  static_cast<const armnn::QLstmDescriptor&>(descriptor);
2181  SerializeQLstmLayer(layer, layerDescriptor, constants, name);
2182  break;
2183  }
2184  case armnn::LayerType::Maximum :
2185  {
2186  SerializeMaximumLayer(layer, name);
2187  break;
2188  }
2189  case armnn::LayerType::Mean :
2190  {
2191  const armnn::MeanDescriptor& layerDescriptor =
2192  static_cast<const armnn::MeanDescriptor&>(descriptor);
2193  SerializeMeanLayer(layer, layerDescriptor, name);
2194  break;
2195  }
2196  case armnn::LayerType::Merge :
2197  {
2198  SerializeMergeLayer(layer, name);
2199  break;
2200  }
2201  case armnn::LayerType::Minimum :
2202  {
2203  SerializeMinimumLayer(layer, name);
2204  break;
2205  }
2206  case armnn::LayerType::Multiplication :
2207  {
2208  SerializeMultiplicationLayer(layer, name);
2209  break;
2210  }
2211  case armnn::LayerType::Normalization :
2212  {
2213  const armnn::NormalizationDescriptor& layerDescriptor =
2214  static_cast<const armnn::NormalizationDescriptor&>(descriptor);
2215  SerializeNormalizationLayer(layer, layerDescriptor, name);
2216  break;
2217  }
2218  case armnn::LayerType::Output:
2219  {
2220  SerializeOutputLayer(layer, id, name);
2221  break;
2222  }
2223  case armnn::LayerType::Pad :
2224  {
2225  const armnn::PadDescriptor& layerDescriptor =
2226  static_cast<const armnn::PadDescriptor&>(descriptor);
2227  SerializePadLayer(layer, layerDescriptor, name);
2228  break;
2229  }
2230  case armnn::LayerType::Permute :
2231  {
2232  const armnn::PermuteDescriptor& layerDescriptor =
2233  static_cast<const armnn::PermuteDescriptor&>(descriptor);
2234  SerializePermuteLayer(layer, layerDescriptor, name);
2235  break;
2236  }
2237  case armnn::LayerType::Pooling2d :
2238  {
2239  const armnn::Pooling2dDescriptor& layerDescriptor =
2240  static_cast<const armnn::Pooling2dDescriptor&>(descriptor);
2241  SerializePooling2dLayer(layer, layerDescriptor, name);
2242  break;
2243  }
2244  case armnn::LayerType::Pooling3d :
2245  {
2246  const armnn::Pooling3dDescriptor& layerDescriptor =
2247  static_cast<const armnn::Pooling3dDescriptor&>(descriptor);
2248  SerializePooling3dLayer(layer, layerDescriptor, name);
2249  break;
2250  }
2251  case armnn::LayerType::Prelu :
2252  {
2253  SerializePreluLayer(layer, name);
2254  break;
2255  }
2256  case armnn::LayerType::Quantize :
2257  {
2258  SerializeQuantizeLayer(layer, name);
2259  break;
2260  }
2261  case armnn::LayerType::QuantizedLstm:
2262  SerializeQuantizedLstmLayer(layer, constants, name);
2263  break;
2264  case armnn::LayerType::Reshape:
2265  {
2266  const armnn::ReshapeDescriptor &layerDescriptor =
2267  static_cast<const armnn::ReshapeDescriptor &>(descriptor);
2268  SerializeReshapeLayer(layer, layerDescriptor, name);
2269  break;
2270  }
2271  case armnn::LayerType::Rank:
2272  {
2273  SerializeRankLayer(layer, name);
2274  break;
2275  }
2276  case armnn::LayerType::Reduce:
2277  {
2278  const armnn::ReduceDescriptor& layerDescriptor =
2279  static_cast<const armnn::ReduceDescriptor&>(descriptor);
2280  SerializeReduceLayer(layer, layerDescriptor, name);
2281  break;
2282  }
2283  case armnn::LayerType::Resize:
2284  {
2285  const armnn::ResizeDescriptor& layerDescriptor =
2286  static_cast<const armnn::ResizeDescriptor&>(descriptor);
2287  SerializeResizeLayer(layer, layerDescriptor, name);
2288  break;
2289  }
2290  case armnn::LayerType::Shape:
2291  {
2292  SerializeShapeLayer(layer, name);
2293  break;
2294  }
2295  case armnn::LayerType::Slice:
2296  {
2297  const armnn::SliceDescriptor& layerDescriptor =
2298  static_cast<const armnn::SliceDescriptor&>(descriptor);
2299  SerializeSliceLayer(layer, layerDescriptor, name);
2300  break;
2301  }
2302  case armnn::LayerType::Softmax:
2303  {
2304  const armnn::SoftmaxDescriptor& layerDescriptor =
2305  static_cast<const armnn::SoftmaxDescriptor&>(descriptor);
2306  SerializeSoftmaxLayer(layer, layerDescriptor, name);
2307  break;
2308  }
2309  case armnn::LayerType::SpaceToBatchNd:
2310  {
2311  const armnn::SpaceToBatchNdDescriptor& layerDescriptor =
2312  static_cast<const armnn::SpaceToBatchNdDescriptor&>(descriptor);
2313  SerializeSpaceToBatchNdLayer(layer, layerDescriptor, name);
2314  break;
2315  }
2316  case armnn::LayerType::SpaceToDepth:
2317  {
2318  const armnn::SpaceToDepthDescriptor& layerDescriptor =
2319  static_cast<const armnn::SpaceToDepthDescriptor&>(descriptor);
2320  SerializeSpaceToDepthLayer(layer, layerDescriptor, name);
2321  break;
2322  }
2323  case armnn::LayerType::Splitter:
2324  {
2325  const armnn::SplitterDescriptor& layerDescriptor =
2326  static_cast<const armnn::SplitterDescriptor&>(descriptor);
2327  SerializeSplitterLayer(layer, layerDescriptor, name);
2328  break;
2329  }
2330  case armnn::LayerType::Stack:
2331  {
2332  const armnn::StackDescriptor& layerDescriptor =
2333  static_cast<const armnn::StackDescriptor&>(descriptor);
2334  SerializeStackLayer(layer, layerDescriptor, name);
2335  break;
2336  }
2337  case armnn::LayerType::StandIn:
2338  {
2339  const armnn::StandInDescriptor& layerDescriptor =
2340  static_cast<const armnn::StandInDescriptor&>(descriptor);
2341  SerializeStandInLayer(layer, layerDescriptor, name);
2342  break;
2343  }
2344  case armnn::LayerType::StridedSlice:
2345  {
2346  const armnn::StridedSliceDescriptor& layerDescriptor =
2347  static_cast<const armnn::StridedSliceDescriptor&>(descriptor);
2348  SerializeStridedSliceLayer(layer, layerDescriptor, name);
2349  break;
2350  }
2351  case armnn::LayerType::Subtraction:
2352  {
2353  SerializeSubtractionLayer(layer, name);
2354  break;
2355  }
2356  case armnn::LayerType::Switch:
2357  {
2358  SerializeSwitchLayer(layer, name);
2359  break;
2360  }
2361  case armnn::LayerType::Transpose:
2362  {
2363  const armnn::TransposeDescriptor& layerDescriptor =
2364  static_cast<const armnn::TransposeDescriptor&>(descriptor);
2365  SerializeTransposeLayer(layer, layerDescriptor, name);
2366  break;
2367  }
2368  case armnn::LayerType::TransposeConvolution2d:
2369  {
2370  const armnn::TransposeConvolution2dDescriptor& layerDescriptor =
2371  static_cast<const armnn::TransposeConvolution2dDescriptor&>(descriptor);
2372  SerializeTransposeConvolution2dLayer(layer, layerDescriptor, constants, name);
2373  break;
2374  }
2375  case armnn::LayerType::UnidirectionalSequenceLstm :
2376  {
2377  const armnn::UnidirectionalSequenceLstmDescriptor& layerDescriptor =
2378  static_cast<const armnn::UnidirectionalSequenceLstmDescriptor&>(descriptor);
2379  SerializeUnidirectionalSequenceLstmLayer(layer, layerDescriptor, constants, name);
2380  break;
2381  }
2382  default:
2383  {
2384  throw InvalidArgumentException(
2385  fmt::format("A layer of unknown type was given to the serializer. Layer name: {}; Layer Id: {}",
2386  layer->GetName(),
2387  id));
2388  }
2389  }
2390 }
armnn::ViewsDescriptor
A ViewsDescriptor for the SplitterLayer.
Definition: Descriptors.hpp:217
armnn::TransposeConvolution2dDescriptor
A TransposeConvolution2dDescriptor for the TransposeConvolution2dLayer.
Definition: Descriptors.hpp:1392
armnn::ReshapeDescriptor
A ReshapeDescriptor for the ReshapeLayer.
Definition: Descriptors.hpp:999
armnn::ComparisonDescriptor
A ComparisonDescriptor for the ComparisonLayer.
Definition: Descriptors.hpp:89
armnn::Convolution2dDescriptor
A Convolution2dDescriptor for the Convolution2dLayer.
Definition: Descriptors.hpp:499
armnn::LogicalBinaryDescriptor
A LogicalBinaryDescriptor for the LogicalBinaryLayer.
Definition: Descriptors.hpp:1463
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
armnn::SpaceToDepthDescriptor
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
Definition: Descriptors.hpp:1051
armnn::BatchToSpaceNdDescriptor
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
Definition: Descriptors.hpp:835
armnn::ResizeDescriptor
A ResizeBilinearDescriptor for the ResizeBilinearLayer.
Definition: Descriptors.hpp:961
armnn::StackDescriptor
A StackDescriptor for the StackLayer.
Definition: Descriptors.hpp:1203
armnn::PadDescriptor
A PadDescriptor for the PadLayer.
Definition: Descriptors.hpp:1148
armnn::LstmDescriptor
An LstmDescriptor for the LstmLayer.
Definition: Descriptors.hpp:1078
armnn::L2NormalizationDescriptor
A L2NormalizationDescriptor for the L2NormalizationLayer.
Definition: Descriptors.hpp:769
armnn::ArgMinMaxDescriptor
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition: Descriptors.hpp:67
armnn::OriginsDescriptor
An OriginsDescriptor for the ConcatLayer.
Definition: Descriptors.hpp:174
armnn::ReduceDescriptor
A ReduceDescriptor for the REDUCE operators.
Definition: Descriptors.hpp:1483
armnn::FullyConnectedDescriptor
A FullyConnectedDescriptor for the FullyConnectedLayer.
Definition: Descriptors.hpp:468
armnn::GatherDescriptor
A GatherDescriptor for the GatherLayer.
Definition: Descriptors.hpp:904
armnn::StandInDescriptor
A StandInDescriptor for the StandIn layer.
Definition: Descriptors.hpp:1233
armnn::QLstmDescriptor
A QLstmDescriptor for the QLstmLayer.
Definition: Descriptors.hpp:1332
armnn::ActivationDescriptor
An ActivationDescriptor for the ActivationLayer.
Definition: Descriptors.hpp:36
armnn::SliceDescriptor
A SliceDescriptor for the SliceLayer.
Definition: Descriptors.hpp:1180
armnn::Convolution3dDescriptor
A Convolution3dDescriptor for the Convolution3dLayer.
Definition: Descriptors.hpp:551
armnn::IConnectableLayer::GetType
virtual LayerType GetType() const =0
Returns the armnn::LayerType of this layer.
armnn::Pooling3dDescriptor
A Pooling3dDescriptor for the Pooling3dLayer.
Definition: Descriptors.hpp:392
armnn::SpaceToBatchNdDescriptor
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
Definition: Descriptors.hpp:1019
armnn::ElementwiseUnaryDescriptor
A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer.
Definition: Descriptors.hpp:109
armnn::MeanDescriptor
A MeanDescriptor for the MeanLayer.
Definition: Descriptors.hpp:1124
armnn::TransposeDescriptor
A TransposeDescriptor for the TransposeLayer.
Definition: Descriptors.hpp:1442
armnn::StridedSliceDescriptor
A StridedSliceDescriptor for the StridedSliceLayer.
Definition: Descriptors.hpp:1255
armnn::IConnectableLayer::GetName
virtual const char * GetName() const =0
Returns the name of the layer.
armnn::Pooling2dDescriptor
A Pooling2dDescriptor for the Pooling2dLayer.
Definition: Descriptors.hpp:332
armnn::NormalizationDescriptor
A NormalizationDescriptor for the NormalizationLayer.
Definition: Descriptors.hpp:729
armnn::DetectionPostProcessDescriptor
Definition: Descriptors.hpp:673
armnn::InstanceNormalizationDescriptor
An InstanceNormalizationDescriptor for InstanceNormalizationLayer.
Definition: Descriptors.hpp:807
armnn::ChannelShuffleDescriptor
A ChannelShuffleDescriptor for the ChannelShuffle operator.
Definition: Descriptors.hpp:1507
armnn::SoftmaxDescriptor
A SoftmaxDescriptor for the SoftmaxLayer.
Definition: Descriptors.hpp:150
armnn::DepthwiseConvolution2dDescriptor
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
Definition: Descriptors.hpp:622
armnn::FillDescriptor
A FillDescriptor for the FillLayer.
Definition: Descriptors.hpp:885
armnn::BatchNormalizationDescriptor
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
Definition: Descriptors.hpp:788
armnn::PermuteDescriptor
A PermuteDescriptor for the PermuteLayer.
Definition: Descriptors.hpp:129

◆ GetFlatBufferBuilder()

flatbuffers::FlatBufferBuilder& GetFlatBufferBuilder ( )
inline

Definition at line 36 of file Serializer.hpp.

37  {
38  return m_flatBufferBuilder;
39  }

◆ GetInputIds()

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

Definition at line 41 of file Serializer.hpp.

42  {
43  return m_inputIds;
44  }

◆ GetOutputIds()

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

Definition at line 46 of file Serializer.hpp.

47  {
48  return m_outputIds;
49  }

◆ GetSerializedLayers()

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

Definition at line 51 of file Serializer.hpp.

References SerializerStrategy::GetVersionTable().

52  {
53  return m_serializedLayers;
54  }

◆ GetVersionTable()

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

Definition at line 1908 of file Serializer.cpp.

References IOutputSlot::CalculateIndexOnOwner(), armnnSerializer::CreateFeatureCompatibilityVersions(), armnnSerializer::CreateInputSlot(), armnnSerializer::CreateOutputSlot(), armnnSerializer::CreateTensorInfo(), IInputSlot::GetConnection(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetNumInputSlots(), IConnectableLayer::GetNumOutputSlots(), IConnectableLayer::GetOutputSlot(), IOutputSlot::GetOwningLayerGuid(), and IOutputSlot::GetTensorInfo().

Referenced by SerializerStrategy::GetSerializedLayers().

1909 {
1910  flatbuffers::Offset<armnnSerializer::FeatureCompatibilityVersions> versionsTable =
1911  serializer::CreateFeatureCompatibilityVersions(
1912  m_flatBufferBuilder,
1913  1, // Binding ids scheme version
1914  1, // Weights layout scheme version
1915  1 // Constant tensors as inputs version
1916  );
1917  return versionsTable;
1918 }
armnnSerializer::CreateFeatureCompatibilityVersions
flatbuffers::Offset< FeatureCompatibilityVersions > CreateFeatureCompatibilityVersions(flatbuffers::FlatBufferBuilder &_fbb, uint32_t bindingIdsScheme=0, uint32_t weightsLayoutScheme=0, uint32_t constantTensorsAsInputs=0)
Definition: ArmnnSchema_generated.h:10896
The documentation for this class was generated from the following files: