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

Method ParsePooling3d

src/armnnDeserializer/Deserializer.cpp:2600–2619  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

2598}
2599
2600void IDeserializer::DeserializerImpl::ParsePooling3d(GraphPtr graph, unsigned int layerIndex)
2601{
2602 CHECK_LAYERS(graph, 0, layerIndex);
2603
2604 auto pooling3dDes = graph->layers()->Get(layerIndex)->layer_as_Pooling3dLayer()->descriptor();
2605 auto inputs = GetInputs(graph, layerIndex);
2606 CHECK_VALID_SIZE(inputs.size(), 1);
2607
2608 auto outputs = GetOutputs(graph, layerIndex);
2609 CHECK_VALID_SIZE(outputs.size(), 1);
2610 auto outputInfo = ToTensorInfo(outputs[0]);
2611
2612 auto pooling3dDescriptor = GetPooling3dDescriptor(pooling3dDes, layerIndex);
2613 auto layerName = GetLayerName(graph, layerIndex);
2614 IConnectableLayer* layer = m_Network->AddPooling3dLayer(pooling3dDescriptor, layerName.c_str());
2615 layer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2616
2617 RegisterInputSlots(graph, layerIndex, layer);
2618 RegisterOutputSlots(graph, layerIndex, layer);
2619}
2620
2621void IDeserializer::DeserializerImpl::ParseQuantize(GraphPtr graph, unsigned int layerIndex)
2622{

Callers

nothing calls this directly

Calls 8

AddPooling3dLayerMethod · 0.80
GetOutputSlotMethod · 0.80
ToTensorInfoFunction · 0.70
GetLayerNameFunction · 0.50
GetMethod · 0.45
sizeMethod · 0.45
c_strMethod · 0.45
SetTensorInfoMethod · 0.45

Tested by

no test coverage detected