Build FlatBuffer for Convolution3dLayer
| 425 | |
| 426 | // Build FlatBuffer for Convolution3dLayer |
| 427 | void SerializerStrategy::SerializeConvolution3dLayer(const armnn::IConnectableLayer* layer, |
| 428 | const armnn::Convolution3dDescriptor& descriptor, |
| 429 | const char* name) |
| 430 | { |
| 431 | IgnoreUnused(name); |
| 432 | |
| 433 | // Create FlatBuffer BaseLayer |
| 434 | auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Convolution3d); |
| 435 | |
| 436 | auto flatBufferDescriptor = CreateConvolution3dDescriptor(m_flatBufferBuilder, |
| 437 | descriptor.m_PadLeft, |
| 438 | descriptor.m_PadRight, |
| 439 | descriptor.m_PadTop, |
| 440 | descriptor.m_PadBottom, |
| 441 | descriptor.m_PadFront, |
| 442 | descriptor.m_PadBack, |
| 443 | descriptor.m_StrideX, |
| 444 | descriptor.m_StrideY, |
| 445 | descriptor.m_StrideZ, |
| 446 | descriptor.m_DilationX, |
| 447 | descriptor.m_DilationY, |
| 448 | descriptor.m_DilationZ, |
| 449 | descriptor.m_BiasEnabled, |
| 450 | GetFlatBufferDataLayout(descriptor.m_DataLayout)); |
| 451 | |
| 452 | // Create the FlatBuffer Convolution3dLayer |
| 453 | auto flatBufferLayer = CreateConvolution3dLayer(m_flatBufferBuilder, |
| 454 | flatBufferBaseLayer, |
| 455 | flatBufferDescriptor); |
| 456 | |
| 457 | // Add the AnyLayer to the FlatBufferLayers |
| 458 | CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_Convolution3dLayer); |
| 459 | } |
| 460 | |
| 461 | void SerializerStrategy::SerializeDepthToSpaceLayer(const armnn::IConnectableLayer* layer, |
| 462 | const armnn::DepthToSpaceDescriptor& descriptor, |
nothing calls this directly
no test coverage detected