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

Method SerializePooling3dLayer

src/armnnSerializer/Serializer.cpp:1142–1173  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1140}
1141
1142void SerializerStrategy::SerializePooling3dLayer(const armnn::IConnectableLayer* layer,
1143 const armnn::Pooling3dDescriptor& pooling3dDescriptor,
1144 const char* name)
1145{
1146 IgnoreUnused(name);
1147
1148 auto fbPooling3dBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Pooling3d);
1149 auto fbPooling3dDescriptor = serializer::CreatePooling3dDescriptor(
1150 m_flatBufferBuilder,
1151 GetFlatBufferPoolingAlgorithm(pooling3dDescriptor.m_PoolType),
1152 pooling3dDescriptor.m_PadLeft,
1153 pooling3dDescriptor.m_PadRight,
1154 pooling3dDescriptor.m_PadTop,
1155 pooling3dDescriptor.m_PadBottom,
1156 pooling3dDescriptor.m_PadFront,
1157 pooling3dDescriptor.m_PadBack,
1158 pooling3dDescriptor.m_PoolWidth,
1159 pooling3dDescriptor.m_PoolHeight,
1160 pooling3dDescriptor.m_PoolDepth,
1161 pooling3dDescriptor.m_StrideX,
1162 pooling3dDescriptor.m_StrideY,
1163 pooling3dDescriptor.m_StrideZ,
1164 GetFlatBufferOutputShapeRounding(pooling3dDescriptor.m_OutputShapeRounding),
1165 GetFlatBufferPaddingMethod(pooling3dDescriptor.m_PaddingMethod),
1166 GetFlatBufferDataLayout(pooling3dDescriptor.m_DataLayout));
1167
1168 auto fbPooling3dLayer = serializer::CreatePooling3dLayer(m_flatBufferBuilder,
1169 fbPooling3dBaseLayer,
1170 fbPooling3dDescriptor);
1171
1172 CreateAnyLayer(fbPooling3dLayer.o, serializer::Layer::Layer_Pooling3dLayer);
1173}
1174
1175void SerializerStrategy::SerializePreluLayer(const armnn::IConnectableLayer* layer,
1176 const char* name)

Callers

nothing calls this directly

Calls 5

GetFlatBufferDataLayoutFunction · 0.85
IgnoreUnusedFunction · 0.50

Tested by

no test coverage detected