| 2598 | } |
| 2599 | |
| 2600 | void 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 | |
| 2621 | void IDeserializer::DeserializerImpl::ParseQuantize(GraphPtr graph, unsigned int layerIndex) |
| 2622 | { |
nothing calls this directly
no test coverage detected