Build FlatBuffer for Convolution2dLayer
| 394 | |
| 395 | // Build FlatBuffer for Convolution2dLayer |
| 396 | void 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 |
| 427 | void SerializerStrategy::SerializeConvolution3dLayer(const armnn::IConnectableLayer* layer, |
nothing calls this directly
no test coverage detected