| 624 | } |
| 625 | |
| 626 | TensorInfo LoadedNetwork::GetInputTensorInfo(LayerBindingId layerId) const |
| 627 | { |
| 628 | for (auto&& inputLayer : m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().GetInputLayers()) |
| 629 | { |
| 630 | if (inputLayer->GetNumOutputSlots() != 1) |
| 631 | { |
| 632 | throw armnn::GraphValidationException("Input layer should have exactly 1 output slot"); |
| 633 | } |
| 634 | |
| 635 | if (inputLayer->GetBindingId() == layerId) |
| 636 | { |
| 637 | return inputLayer->GetOutputSlot(0).GetTensorInfo(); |
| 638 | } |
| 639 | } |
| 640 | |
| 641 | throw InvalidArgumentException(fmt::format("No input layer is associated with id {}", layerId)); |
| 642 | } |
| 643 | |
| 644 | TensorInfo LoadedNetwork::GetOutputTensorInfo(LayerBindingId layerId) const |
| 645 | { |
nothing calls this directly
no test coverage detected