| 642 | } |
| 643 | |
| 644 | TensorInfo LoadedNetwork::GetOutputTensorInfo(LayerBindingId layerId) const |
| 645 | { |
| 646 | for (auto&& outputLayer : m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().GetOutputLayers()) |
| 647 | { |
| 648 | if (outputLayer->GetNumInputSlots() != 1) |
| 649 | { |
| 650 | throw armnn::GraphValidationException("Output layer should have exactly 1 input slot"); |
| 651 | } |
| 652 | |
| 653 | if (!outputLayer->GetInputSlot(0).GetConnection()) |
| 654 | { |
| 655 | throw armnn::GraphValidationException("Input slot on Output layer must be connected"); |
| 656 | } |
| 657 | |
| 658 | if (outputLayer->GetBindingId() == layerId) |
| 659 | { |
| 660 | return outputLayer->GetInputSlot(0).GetTensorInfo(); |
| 661 | } |
| 662 | } |
| 663 | |
| 664 | throw InvalidArgumentException(fmt::format("No output layer is associated with id {}", layerId)); |
| 665 | } |
| 666 | |
| 667 | const IWorkloadFactory& LoadedNetwork::GetWorkloadFactory(const Layer& layer) const |
| 668 | { |
nothing calls this directly
no test coverage detected