| 2619 | } |
| 2620 | |
| 2621 | void IDeserializer::DeserializerImpl::ParseQuantize(GraphPtr graph, unsigned int layerIndex) |
| 2622 | { |
| 2623 | CHECK_LAYERS(graph, 0, layerIndex); |
| 2624 | |
| 2625 | auto inputs = GetInputs(graph, layerIndex); |
| 2626 | CHECK_VALID_SIZE(inputs.size(), 1); |
| 2627 | |
| 2628 | auto outputs = GetOutputs(graph, layerIndex); |
| 2629 | CHECK_VALID_SIZE(outputs.size(), 1); |
| 2630 | auto outputInfo = ToTensorInfo(outputs[0]); |
| 2631 | |
| 2632 | auto layerName = GetLayerName(graph, layerIndex); |
| 2633 | IConnectableLayer* layer = m_Network->AddQuantizeLayer(layerName.c_str()); |
| 2634 | layer->GetOutputSlot(0).SetTensorInfo(outputInfo); |
| 2635 | |
| 2636 | RegisterInputSlots(graph, layerIndex, layer); |
| 2637 | RegisterOutputSlots(graph, layerIndex, layer); |
| 2638 | } |
| 2639 | |
| 2640 | armnn::TensorInfo IDeserializer::DeserializerImpl::OutputShapeOfReshape(const armnn::TensorInfo& inputTensorInfo, |
| 2641 | const std::vector<uint32_t>& targetDimsIn) |
nothing calls this directly
no test coverage detected