diff options
Diffstat (limited to 'src/armnn/layers/OutputLayer.cpp')
| -rw-r--r-- | src/armnn/layers/OutputLayer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/layers/OutputLayer.cpp b/src/armnn/layers/OutputLayer.cpp index f00e0a5259..5ff91880f8 100644 --- a/src/armnn/layers/OutputLayer.cpp +++ b/src/armnn/layers/OutputLayer.cpp @@ -29,8 +29,10 @@ OutputLayer* OutputLayer::Clone(Graph& graph) const return CloneBase<OutputLayer>(graph, GetBindingId(), GetName()); } -void OutputLayer::ValidateTensorShapesFromInputs() +void OutputLayer::ValidateTensorShapesFromInputs(ShapeInferenceMethod shapeInferenceMethod) { + IgnoreUnused(shapeInferenceMethod); + // Just validates that the input is connected. ConditionalThrow<LayerValidationException>(GetInputSlot(0).GetConnection() != nullptr, "OutputLayer: Input slot must be connected."); |