MCPcopy Create free account
hub / github.com/ARM-software/armnn / SerializeConvolution2dLayer

Method SerializeConvolution2dLayer

src/armnnSerializer/Serializer.cpp:396–424  ·  view source on GitHub ↗

Build FlatBuffer for Convolution2dLayer

Source from the content-addressed store, hash-verified

394
395// Build FlatBuffer for Convolution2dLayer
396void SerializerStrategy::SerializeConvolution2dLayer(const armnn::IConnectableLayer* layer,
397 const armnn::Convolution2dDescriptor& descriptor,
398 const char* name)
399{
400 IgnoreUnused(name);
401
402 // Create FlatBuffer BaseLayer
403 auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Convolution2d);
404
405 auto flatBufferDescriptor = CreateConvolution2dDescriptor(m_flatBufferBuilder,
406 descriptor.m_PadLeft,
407 descriptor.m_PadRight,
408 descriptor.m_PadTop,
409 descriptor.m_PadBottom,
410 descriptor.m_StrideX,
411 descriptor.m_StrideY,
412 descriptor.m_DilationX,
413 descriptor.m_DilationY,
414 descriptor.m_BiasEnabled,
415 GetFlatBufferDataLayout(descriptor.m_DataLayout));
416
417 // Create the FlatBuffer Convolution2dLayer
418 auto flatBufferLayer = CreateConvolution2dLayer(m_flatBufferBuilder,
419 flatBufferBaseLayer,
420 flatBufferDescriptor);
421
422 // Add the AnyLayer to the FlatBufferLayers
423 CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_Convolution2dLayer);
424}
425
426// Build FlatBuffer for Convolution3dLayer
427void SerializerStrategy::SerializeConvolution3dLayer(const armnn::IConnectableLayer* layer,

Callers

nothing calls this directly

Calls 2

GetFlatBufferDataLayoutFunction · 0.85
IgnoreUnusedFunction · 0.50

Tested by

no test coverage detected