Lines Matching full:layer
114 // Build FlatBuffer for Input Layer
115 void SerializerStrategy::SerializeInputLayer(const armnn::IConnectableLayer* layer, LayerBindingId … in SerializeInputLayer() argument
120 auto flatBufferInputBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Input); in SerializeInputLayer()
126 // Push layer binding id to outputIds. in SerializeInputLayer()
133 CreateAnyLayer(flatBufferInputLayer.o, serializer::Layer::Layer_InputLayer); in SerializeInputLayer()
136 // Build FlatBuffer for Output Layer
137 void SerializerStrategy::SerializeOutputLayer(const armnn::IConnectableLayer* layer, in SerializeOutputLayer() argument
143 … auto flatBufferOutputBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Output); in SerializeOutputLayer()
149 // Push layer binding id to outputIds. in SerializeOutputLayer()
155 CreateAnyLayer(flatBufferOutputLayer.o, serializer::Layer::Layer_OutputLayer); in SerializeOutputLayer()
158 // Build FlatBuffer for Activation Layer
159 void SerializerStrategy::SerializeActivationLayer(const armnn::IConnectableLayer* layer, in SerializeActivationLayer() argument
166 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Activation); in SerializeActivationLayer()
180 CreateAnyLayer(flatBufferAdditionLayer.o, serializer::Layer::Layer_ActivationLayer); in SerializeActivationLayer()
183 // Build FlatBuffer for Addition Layer
184 void SerializerStrategy::SerializeAdditionLayer(const armnn::IConnectableLayer* layer, const char* … in SerializeAdditionLayer() argument
189 …auto flatBufferAdditionBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Additio… in SerializeAdditionLayer()
195 CreateAnyLayer(flatBufferAdditionLayer.o, serializer::Layer::Layer_AdditionLayer); in SerializeAdditionLayer()
198 // Build FlatBuffer for ArgMinMax Layer
199 void SerializerStrategy::SerializeArgMinMaxLayer(const armnn::IConnectableLayer *layer, in SerializeArgMinMaxLayer() argument
206 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_ArgMinMax); in SerializeArgMinMaxLayer()
218 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ArgMinMaxLayer); in SerializeArgMinMaxLayer()
221 void SerializerStrategy::SerializeBatchMatMulLayer(const armnn::IConnectableLayer* layer, in SerializeBatchMatMulLayer() argument
228 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_BatchMatMul); in SerializeBatchMatMulLayer()
245 CreateAnyLayer(flatBufferBatchMatMulLayer.o, serializer::Layer::Layer_BatchMatMulLayer); in SerializeBatchMatMulLayer()
248 // Build FlatBuffer for BatchToSpaceNd Layer
249 void SerializerStrategy::SerializeBatchToSpaceNdLayer(const armnn::IConnectableLayer* layer, in SerializeBatchToSpaceNdLayer() argument
256 …auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_BatchToSpaceNd); in SerializeBatchToSpaceNdLayer()
276 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_BatchToSpaceNdLayer); in SerializeBatchToSpaceNdLayer()
280 const armnn::IConnectableLayer* layer, in SerializeBatchNormalizationLayer() argument
292 …auto fbBatchNormalizationBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Batc… in SerializeBatchNormalizationLayer()
310 CreateAnyLayer(fbBatchNormalizationLayer.o, serializer::Layer::Layer_BatchNormalizationLayer); in SerializeBatchNormalizationLayer()
313 void SerializerStrategy::SerializeCastLayer(const armnn::IConnectableLayer* layer, in SerializeCastLayer() argument
318 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Cast); in SerializeCastLayer()
320 CreateAnyLayer(fbCastLayer.o, serializer::Layer::Layer_CastLayer); in SerializeCastLayer()
323 void SerializerStrategy::SerializeChannelShuffleLayer(const armnn::IConnectableLayer* layer, in SerializeChannelShuffleLayer() argument
331 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_ChannelShuffle); in SerializeChannelShuffleLayer()
333 CreateAnyLayer(fbChannelShuffleLayer.o, serializer::Layer::Layer_ChannelShuffleLayer); in SerializeChannelShuffleLayer()
336 void SerializerStrategy::SerializeComparisonLayer(const armnn::IConnectableLayer* layer, in SerializeComparisonLayer() argument
342 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Comparison); in SerializeComparisonLayer()
348 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_ComparisonLayer); in SerializeComparisonLayer()
351 // Build FlatBuffer for Constant Layer
352 void SerializerStrategy::SerializeConstantLayer(const armnn::IConnectableLayer* layer, in SerializeConstantLayer() argument
361 …auto flatBufferConstantBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Constan… in SerializeConstantLayer()
371 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ConstantLayer); in SerializeConstantLayer()
375 void SerializerStrategy::SerializeConvolution2dLayer(const armnn::IConnectableLayer* layer, in SerializeConvolution2dLayer() argument
382 … auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Convolution2d); in SerializeConvolution2dLayer()
402 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_Convolution2dLayer); in SerializeConvolution2dLayer()
406 void SerializerStrategy::SerializeConvolution3dLayer(const armnn::IConnectableLayer* layer, in SerializeConvolution3dLayer() argument
413 … auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Convolution3d); in SerializeConvolution3dLayer()
437 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_Convolution3dLayer); in SerializeConvolution3dLayer()
440 void SerializerStrategy::SerializeDepthToSpaceLayer(const armnn::IConnectableLayer* layer, in SerializeDepthToSpaceLayer() argument
446 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_DepthToSpace); in SerializeDepthToSpaceLayer()
453 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_DepthToSpaceLayer); in SerializeDepthToSpaceLayer()
456 void SerializerStrategy::SerializeDepthwiseConvolution2dLayer(const armnn::IConnectableLayer* layer, in SerializeDepthwiseConvolution2dLayer() argument
462 …auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_DepthwiseConvolution2d… in SerializeDepthwiseConvolution2dLayer()
479 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_DepthwiseConvolution2dLayer); in SerializeDepthwiseConvolution2dLayer()
482 void SerializerStrategy::SerializeDequantizeLayer(const armnn::IConnectableLayer* layer, in SerializeDequantizeLayer() argument
487 … auto fbDequantizeBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Dequantize); in SerializeDequantizeLayer()
490 CreateAnyLayer(fbDequantizeLayer.o, serializer::Layer::Layer_DequantizeLayer); in SerializeDequantizeLayer()
493 void SerializerStrategy::SerializeDetectionPostProcessLayer(const armnn::IConnectableLayer* layer, in SerializeDetectionPostProcessLayer() argument
502 … auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_DetectionPostProcess); in SerializeDetectionPostProcessLayer()
523 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_DetectionPostProcessLayer); in SerializeDetectionPostProcessLayer()
526 void SerializerStrategy::SerializeDivisionLayer(const armnn::IConnectableLayer* layer, const char* … in SerializeDivisionLayer() argument
530 auto fbDivisionBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Division); in SerializeDivisionLayer()
533 CreateAnyLayer(fbDivisionLayer.o, serializer::Layer::Layer_DivisionLayer); in SerializeDivisionLayer()
536 void SerializerStrategy::SerializeElementwiseBinaryLayer(const armnn::IConnectableLayer* layer, in SerializeElementwiseBinaryLayer() argument
542 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_ElementwiseBinary); in SerializeElementwiseBinaryLayer()
548 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_ElementwiseBinaryLayer); in SerializeElementwiseBinaryLayer()
551 void SerializerStrategy::SerializeElementwiseUnaryLayer(const armnn::IConnectableLayer* layer, in SerializeElementwiseUnaryLayer() argument
557 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_ElementwiseUnary); in SerializeElementwiseUnaryLayer()
563 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_ElementwiseUnaryLayer); in SerializeElementwiseUnaryLayer()
566 void SerializerStrategy::SerializeFillLayer(const armnn::IConnectableLayer* layer, in SerializeFillLayer() argument
572 auto fbFillBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Fill); in SerializeFillLayer()
578 CreateAnyLayer(fbFillLayer.o, serializer::Layer::Layer_FillLayer); in SerializeFillLayer()
581 void SerializerStrategy::SerializeFloorLayer(const armnn::IConnectableLayer *layer, const char *nam… in SerializeFloorLayer() argument
585 auto flatBufferFloorBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Floor); in SerializeFloorLayer()
588 CreateAnyLayer(flatBufferFloorLayer.o, serializer::Layer::Layer_FloorLayer); in SerializeFloorLayer()
591 void SerializerStrategy::SerializeGatherLayer(const armnn::IConnectableLayer* layer, in SerializeGatherLayer() argument
599 auto fbGatherBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Gather); in SerializeGatherLayer()
602 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_GatherLayer); in SerializeGatherLayer()
605 void SerializerStrategy::SerializeGatherNdLayer(const armnn::IConnectableLayer* layer, in SerializeGatherNdLayer() argument
610 auto fbGatherNdBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_GatherNd); in SerializeGatherNdLayer()
613 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_GatherNdLayer); in SerializeGatherNdLayer()
617 const armnn::IConnectableLayer* layer, in SerializeInstanceNormalizationLayer() argument
630 … auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_InstanceNormalization); in SerializeInstanceNormalizationLayer()
633 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_InstanceNormalizationLayer); in SerializeInstanceNormalizationLayer()
636 void SerializerStrategy::SerializeL2NormalizationLayer(const armnn::IConnectableLayer* layer, in SerializeL2NormalizationLayer() argument
643 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_L2Normalization); in SerializeL2NormalizationLayer()
651 // Create FlatBuffer layer in SerializeL2NormalizationLayer()
654 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_L2NormalizationLayer); in SerializeL2NormalizationLayer()
657 void SerializerStrategy::SerializeLogicalBinaryLayer(const armnn::IConnectableLayer* layer, in SerializeLogicalBinaryLayer() argument
663 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_LogicalBinary); in SerializeLogicalBinaryLayer()
669 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_LogicalBinaryLayer); in SerializeLogicalBinaryLayer()
672 void SerializerStrategy::SerializeLogSoftmaxLayer(const armnn::IConnectableLayer* layer, in SerializeLogSoftmaxLayer() argument
679 …auto flatBufferLogSoftmaxBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_LogSo… in SerializeLogSoftmaxLayer()
693 CreateAnyLayer(flatBufferLogSoftmaxLayer.o, serializer::Layer::Layer_LogSoftmaxLayer); in SerializeLogSoftmaxLayer()
696 void SerializerStrategy::SerializeLstmLayer(const armnn::IConnectableLayer* layer, in SerializeLstmLayer() argument
703 auto fbLstmBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Lstm); in SerializeLstmLayer()
809 CreateAnyLayer(fbLstmLayer.o, serializer::Layer::Layer_LstmLayer); in SerializeLstmLayer()
812 void SerializerStrategy::SerializeMaximumLayer(const armnn::IConnectableLayer* layer, const char* n… in SerializeMaximumLayer() argument
816 auto fbMaximumBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Maximum); in SerializeMaximumLayer()
819 CreateAnyLayer(fbMaximumLayer.o, serializer::Layer::Layer_MaximumLayer); in SerializeMaximumLayer()
822 void SerializerStrategy::SerializeMeanLayer(const armnn::IConnectableLayer* layer, in SerializeMeanLayer() argument
828 auto fbMeanBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Mean); in SerializeMeanLayer()
837 CreateAnyLayer(fbMeanLayer.o, serializer::Layer::Layer_MeanLayer); in SerializeMeanLayer()
840 void SerializerStrategy::SerializeMinimumLayer(const armnn::IConnectableLayer* layer, const char* n… in SerializeMinimumLayer() argument
844 auto fbMinimumBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Minimum); in SerializeMinimumLayer()
847 CreateAnyLayer(fbMinimumLayer.o, serializer::Layer::Layer_MinimumLayer); in SerializeMinimumLayer()
850 void SerializerStrategy::SerializeMergeLayer(const armnn::IConnectableLayer* layer, const char* nam… in SerializeMergeLayer() argument
854 auto fbMergeBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Merge); in SerializeMergeLayer()
857 CreateAnyLayer(fbMergeLayer.o, serializer::Layer::Layer_MergeLayer); in SerializeMergeLayer()
860 void SerializerStrategy::SerializeConcatLayer(const armnn::IConnectableLayer* layer, in SerializeConcatLayer() argument
866 … auto flatBufferConcatBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Concat); in SerializeConcatLayer()
892 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ConcatLayer); in SerializeConcatLayer()
895 void SerializerStrategy::SerializeMultiplicationLayer(const armnn::IConnectableLayer* layer, const … in SerializeMultiplicationLayer() argument
899 …auto fbMultiplicationBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Multiplic… in SerializeMultiplicationLayer()
903 CreateAnyLayer(fbMultiplicationLayer.o, serializer::Layer::Layer_MultiplicationLayer); in SerializeMultiplicationLayer()
906 void SerializerStrategy::SerializePadLayer(const armnn::IConnectableLayer* layer, in SerializePadLayer() argument
912 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Pad); in SerializePadLayer()
930 CreateAnyLayer(flatBufferPadLayer.o, serializer::Layer::Layer_PadLayer); in SerializePadLayer()
933 void SerializerStrategy::SerializePermuteLayer(const armnn::IConnectableLayer* layer, in SerializePermuteLayer() argument
940 …auto flatBufferPermuteBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Permute); in SerializePermuteLayer()
957 CreateAnyLayer(flatBufferPermuteLayer.o, serializer::Layer::Layer_PermuteLayer); in SerializePermuteLayer()
960 // Build FlatBuffer for Rank Layer
961 void SerializerStrategy::SerializeRankLayer(const armnn::IConnectableLayer* layer, in SerializeRankLayer() argument
965 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Rank); in SerializeRankLayer()
968 CreateAnyLayer(flatBufferRankLayer.o, serializer::Layer::Layer_RankLayer); in SerializeRankLayer()
971 void SerializerStrategy::SerializeReduceLayer(const armnn::IConnectableLayer* layer, in SerializeReduceLayer() argument
975 auto fbReduceBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Reduce); in SerializeReduceLayer()
984 CreateAnyLayer(fbReduceLayer.o, serializer::Layer::Layer_ReduceLayer); in SerializeReduceLayer()
987 // Build FlatBuffer for Reshape Layer
988 void SerializerStrategy::SerializeReshapeLayer(const armnn::IConnectableLayer* layer, in SerializeReshapeLayer() argument
995 …auto flatBufferReshapeBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Reshape); in SerializeReshapeLayer()
1011 CreateAnyLayer(flatBufferReshapeLayer.o, serializer::Layer::Layer_ReshapeLayer); in SerializeReshapeLayer()
1014 void SerializerStrategy::SerializeResizeLayer(const armnn::IConnectableLayer* layer, in SerializeResizeLayer() argument
1020 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Resize); in SerializeResizeLayer()
1035 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ResizeLayer); in SerializeResizeLayer()
1038 void SerializerStrategy::SerializeSliceLayer(const armnn::IConnectableLayer* layer, in SerializeSliceLayer() argument
1044 auto fbSliceBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Slice); in SerializeSliceLayer()
1051 CreateAnyLayer(fbSliceLayer.o, serializer::Layer::Layer_SliceLayer); in SerializeSliceLayer()
1054 // Build FlatBuffer for Softmax Layer
1055 void SerializerStrategy::SerializeSoftmaxLayer(const armnn::IConnectableLayer* layer, in SerializeSoftmaxLayer() argument
1062 …auto flatBufferSoftmaxBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Softmax); in SerializeSoftmaxLayer()
1076 CreateAnyLayer(flatBufferSoftmaxLayer.o, serializer::Layer::Layer_SoftmaxLayer); in SerializeSoftmaxLayer()
1079 void SerializerStrategy::SerializePooling2dLayer(const armnn::IConnectableLayer* layer, in SerializePooling2dLayer() argument
1085 auto fbPooling2dBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Pooling2d); in SerializePooling2dLayer()
1105 CreateAnyLayer(fbPooling2dLayer.o, serializer::Layer::Layer_Pooling2dLayer); in SerializePooling2dLayer()
1108 void SerializerStrategy::SerializePooling3dLayer(const armnn::IConnectableLayer* layer, in SerializePooling3dLayer() argument
1114 auto fbPooling3dBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Pooling3d); in SerializePooling3dLayer()
1138 CreateAnyLayer(fbPooling3dLayer.o, serializer::Layer::Layer_Pooling3dLayer); in SerializePooling3dLayer()
1141 void SerializerStrategy::SerializePreluLayer(const armnn::IConnectableLayer* layer, in SerializePreluLayer() argument
1147 auto flatBufferPreluBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Prelu); in SerializePreluLayer()
1153 CreateAnyLayer(flatBufferPreluLayer.o, serializer::Layer::Layer_PreluLayer); in SerializePreluLayer()
1156 void SerializerStrategy::SerializeQuantizeLayer(const armnn::IConnectableLayer *layer, const char *… in SerializeQuantizeLayer() argument
1160 auto fbQuantizeBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Quantize); in SerializeQuantizeLayer()
1163 CreateAnyLayer(fbQuantizeLayer.o, serializer::Layer::Layer_QuantizeLayer); in SerializeQuantizeLayer()
1166 // Build FlatBuffer for FullyConnected Layer
1167 void SerializerStrategy::SerializeFullyConnectedLayer(const armnn::IConnectableLayer* layer, in SerializeFullyConnectedLayer() argument
1172 …auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_FullyConnected); in SerializeFullyConnectedLayer()
1187 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_FullyConnectedLayer); in SerializeFullyConnectedLayer()
1190 // Build FlatBuffer for SpaceToBatchNd Layer
1191 void SerializerStrategy::SerializeSpaceToBatchNdLayer(const armnn::IConnectableLayer* layer, in SerializeSpaceToBatchNdLayer() argument
1198 …auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_SpaceToBatchNd); in SerializeSpaceToBatchNdLayer()
1218 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_SpaceToBatchNdLayer); in SerializeSpaceToBatchNdLayer()
1222 void SerializerStrategy::SerializeSpaceToDepthLayer(const armnn::IConnectableLayer* layer, in SerializeSpaceToDepthLayer() argument
1228 … auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_SpaceToDepth); in SerializeSpaceToDepthLayer()
1238 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_SpaceToDepthLayer); in SerializeSpaceToDepthLayer()
1241 // Build FlatBuffer for Splitter Layer
1242 void SerializerStrategy::SerializeSplitterLayer(const armnn::IConnectableLayer* layer, in SerializeSplitterLayer() argument
1299 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Splitter); in SerializeSplitterLayer()
1305 CreateAnyLayer(flatBufferSplitterLayer.o, serializer::Layer::Layer_SplitterLayer); in SerializeSplitterLayer()
1308 void SerializerStrategy::SerializeNormalizationLayer(const armnn::IConnectableLayer* layer, in SerializeNormalizationLayer() argument
1314 …auto fbNormalizationBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Normaliza… in SerializeNormalizationLayer()
1330 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_NormalizationLayer); in SerializeNormalizationLayer()
1333 void SerializerStrategy::SerializeShapeLayer(const armnn::IConnectableLayer* layer, in SerializeShapeLayer() argument
1338 auto shapeBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Shape); in SerializeShapeLayer()
1341 CreateAnyLayer(shapeLayer.o, serializer::Layer::Layer_ShapeLayer); in SerializeShapeLayer()
1344 void SerializerStrategy::SerializeStackLayer(const armnn::IConnectableLayer* layer, in SerializeStackLayer() argument
1350 auto stackBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Stack); in SerializeStackLayer()
1364 CreateAnyLayer(stackLayer.o, serializer::Layer::Layer_StackLayer); in SerializeStackLayer()
1367 void SerializerStrategy::SerializeStandInLayer(const armnn::IConnectableLayer *layer, in SerializeStandInLayer() argument
1377 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_StandIn); in SerializeStandInLayer()
1380 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_StandInLayer); in SerializeStandInLayer()
1383 void SerializerStrategy::SerializeStridedSliceLayer(const armnn::IConnectableLayer* layer, in SerializeStridedSliceLayer() argument
1389 … auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_StridedSlice); in SerializeStridedSliceLayer()
1407 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_StridedSliceLayer); in SerializeStridedSliceLayer()
1410 void SerializerStrategy::SerializeSubtractionLayer(const armnn::IConnectableLayer* layer, const cha… in SerializeSubtractionLayer() argument
1414 …auto fbSubtractionBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Subtraction); in SerializeSubtractionLayer()
1417 CreateAnyLayer(fbSubtractionLayer.o, serializer::Layer::Layer_SubtractionLayer); in SerializeSubtractionLayer()
1420 void SerializerStrategy::SerializeSwitchLayer(const armnn::IConnectableLayer* layer, const char* na… in SerializeSwitchLayer() argument
1424 auto fbSwitchBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Switch); in SerializeSwitchLayer()
1427 CreateAnyLayer(fbSwitchLayer.o, serializer::Layer::Layer_SwitchLayer); in SerializeSwitchLayer()
1431 const armnn::IConnectableLayer* layer, in SerializeTransposeConvolution2dLayer() argument
1440 auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Convolution2d); in SerializeTransposeConvolution2dLayer()
1466 CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_TransposeConvolution2dLayer); in SerializeTransposeConvolution2dLayer()
1469 void SerializerStrategy::SerializeTransposeLayer(const armnn::IConnectableLayer* layer, in SerializeTransposeLayer() argument
1476 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Transpose); in SerializeTransposeLayer()
1493 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_TransposeLayer); in SerializeTransposeLayer()
1496 void SerializerStrategy::SerializeQLstmLayer(const armnn::IConnectableLayer* layer, in SerializeQLstmLayer() argument
1503 auto fbQLstmBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_QLstm); in SerializeQLstmLayer()
1572 // Layer norm in SerializeQLstmLayer()
1619 CreateAnyLayer(fbQLstmLayer.o, serializer::Layer::Layer_QLstmLayer); in SerializeQLstmLayer()
1622 void SerializerStrategy::SerializeQuantizedLstmLayer(const armnn::IConnectableLayer* layer, in SerializeQuantizedLstmLayer() argument
1628 …auto fbQuantizedLstmBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_QuantizedL… in SerializeQuantizedLstmLayer()
1669 CreateAnyLayer(fbQuantizedLstmLayer.o, serializer::Layer::Layer_QuantizedLstmLayer); in SerializeQuantizedLstmLayer()
1673 const armnn::IConnectableLayer* layer, in SerializeUnidirectionalSequenceLstmLayer() argument
1681 CreateLayerBase(layer, serializer::LayerType::LayerType_UnidirectionalSequenceLstm); in SerializeUnidirectionalSequenceLstmLayer()
1786 …CreateAnyLayer(fbUnidirectionalSequenceLstmLayer.o, serializer::Layer::Layer_UnidirectionalSequenc… in SerializeUnidirectionalSequenceLstmLayer()
1789 …::Offset<serializer::LayerBase> SerializerStrategy::CreateLayerBase(const IConnectableLayer* layer, in CreateLayerBase() argument
1793 uint32_t fbIndex = GetSerializedId(layer->GetGuid()); in CreateLayerBase()
1795 std::vector<fb::Offset<serializer::InputSlot>> inputSlots = CreateInputSlots(layer); in CreateLayerBase()
1796 std::vector<fb::Offset<serializer::OutputSlot>> outputSlots = CreateOutputSlots(layer); in CreateLayerBase()
1800 m_flatBufferBuilder.CreateString(layer->GetName()), in CreateLayerBase()
1806 …zerStrategy::CreateAnyLayer(const flatbuffers::Offset<void>& layer, const serializer::Layer serial… in CreateAnyLayer() argument
1809 auto anyLayer = armnnSerializer::CreateAnyLayer(m_flatBufferBuilder, serializerLayer, layer); in CreateAnyLayer()
1947 SerializerStrategy::CreateInputSlots(const armnn::IConnectableLayer* layer) in CreateInputSlots() argument
1952 for (unsigned int slotIndex = 0; slotIndex<layer->GetNumInputSlots(); ++slotIndex) in CreateInputSlots()
1954 const IInputSlot& inputSlot = layer->GetInputSlot(slotIndex); in CreateInputSlots()
1969 SerializerStrategy::CreateOutputSlots(const armnn::IConnectableLayer* layer) in CreateOutputSlots() argument
1974 for (unsigned int slotIndex = 0; slotIndex < layer->GetNumOutputSlots(); ++slotIndex) in CreateOutputSlots()
1976 const IOutputSlot& outputSlot = layer->GetOutputSlot(slotIndex); in CreateOutputSlots()
1987 void SerializerStrategy::ExecuteStrategy(const armnn::IConnectableLayer* layer, in ExecuteStrategy() argument
1995 switch (layer->GetType()) in ExecuteStrategy()
2001 SerializeActivationLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2006 SerializeAdditionLayer(layer, name); in ExecuteStrategy()
2013 SerializeArgMinMaxLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2020 SerializeBatchMatMulLayer(layer, in ExecuteStrategy()
2029 SerializeBatchNormalizationLayer(layer, in ExecuteStrategy()
2039 SerializeBatchToSpaceNdLayer(layer, in ExecuteStrategy()
2046 SerializeCastLayer(layer, name); in ExecuteStrategy()
2053 SerializeChannelShuffleLayer(layer, in ExecuteStrategy()
2062 SerializeComparisonLayer(layer, in ExecuteStrategy()
2071 SerializeConcatLayer(layer, in ExecuteStrategy()
2078 SerializeConstantLayer(layer, in ExecuteStrategy()
2087 SerializeConvolution2dLayer(layer, in ExecuteStrategy()
2096 SerializeConvolution3dLayer(layer, in ExecuteStrategy()
2105 SerializeDepthToSpaceLayer(layer, in ExecuteStrategy()
2114 SerializeDepthwiseConvolution2dLayer(layer, in ExecuteStrategy()
2121 SerializeDequantizeLayer(layer, in ExecuteStrategy()
2129 SerializeDetectionPostProcessLayer(layer, layerDescriptor, constants, name); in ExecuteStrategy()
2134 SerializeDivisionLayer(layer, name); in ExecuteStrategy()
2141 SerializeElementwiseBinaryLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2148 SerializeElementwiseUnaryLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2155 SerializeFillLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2160 SerializeFloorLayer(layer, name); in ExecuteStrategy()
2167 SerializeFullyConnectedLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2174 SerializeGatherLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2179 SerializeGatherNdLayer(layer, name); in ExecuteStrategy()
2184 SerializeInputLayer(layer, id, name); in ExecuteStrategy()
2191 SerializeInstanceNormalizationLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2198 SerializeL2NormalizationLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2205 SerializeLogicalBinaryLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2212 SerializeLogSoftmaxLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2219 SerializeLstmLayer(layer, layerDescriptor, constants, name); in ExecuteStrategy()
2226 SerializeQLstmLayer(layer, layerDescriptor, constants, name); in ExecuteStrategy()
2231 SerializeMaximumLayer(layer, name); in ExecuteStrategy()
2238 SerializeMeanLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2243 SerializeMergeLayer(layer, name); in ExecuteStrategy()
2248 SerializeMinimumLayer(layer, name); in ExecuteStrategy()
2253 SerializeMultiplicationLayer(layer, name); in ExecuteStrategy()
2260 SerializeNormalizationLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2265 SerializeOutputLayer(layer, id, name); in ExecuteStrategy()
2272 SerializePadLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2279 SerializePermuteLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2286 SerializePooling2dLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2293 SerializePooling3dLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2298 SerializePreluLayer(layer, name); in ExecuteStrategy()
2303 SerializeQuantizeLayer(layer, name); in ExecuteStrategy()
2307 SerializeQuantizedLstmLayer(layer, constants, name); in ExecuteStrategy()
2313 SerializeReshapeLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2318 SerializeRankLayer(layer, name); in ExecuteStrategy()
2325 SerializeReduceLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2332 SerializeResizeLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2337 SerializeShapeLayer(layer, name); in ExecuteStrategy()
2344 SerializeSliceLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2351 SerializeSoftmaxLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2358 SerializeSpaceToBatchNdLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2365 SerializeSpaceToDepthLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2372 SerializeSplitterLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2379 SerializeStackLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2386 SerializeStandInLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2393 SerializeStridedSliceLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2398 SerializeSubtractionLayer(layer, name); in ExecuteStrategy()
2403 SerializeSwitchLayer(layer, name); in ExecuteStrategy()
2410 SerializeTransposeLayer(layer, layerDescriptor, name); in ExecuteStrategy()
2417 SerializeTransposeConvolution2dLayer(layer, layerDescriptor, constants, name); in ExecuteStrategy()
2424 SerializeUnidirectionalSequenceLstmLayer(layer, layerDescriptor, constants, name); in ExecuteStrategy()
2430 … fmt::format("A layer of unknown type was given to the serializer. Layer name: {}; Layer Id: {}", in ExecuteStrategy()
2431 layer->GetName(), in ExecuteStrategy()