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

Function GetFlatBufferPoolingAlgorithm

src/armnnSerializer/SerializerUtils.cpp:164–176  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

162}
163
164armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgorithm)
165{
166 switch (poolingAlgorithm)
167 {
168 case armnn::PoolingAlgorithm::Average:
169 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average;
170 case armnn::PoolingAlgorithm::L2:
171 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2;
172 case armnn::PoolingAlgorithm::Max:
173 default:
174 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max;
175 }
176}
177
178armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding(armnn::OutputShapeRounding outputShapeRounding)
179{

Callers 2

Calls

no outgoing calls

Tested by

no test coverage detected