diff options
Diffstat (limited to 'src/armnn/Graph.cpp')
-rw-r--r-- | src/armnn/Graph.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp index bd0cb34d57..cc3384748a 100644 --- a/src/armnn/Graph.cpp +++ b/src/armnn/Graph.cpp @@ -489,7 +489,7 @@ void Graph::EraseSubgraphLayers(SubgraphView &subgraph) subgraph.Clear(); } -void Graph::InferTensorInfos() +void Graph::InferTensorInfos(ShapeInferenceMethod shapeInferenceMethod) { for (auto&& layer : TopologicalSort()) { @@ -512,7 +512,7 @@ void Graph::InferTensorInfos() throw LayerValidationException("All inputs must have the TensorInfo set at this point."); } } - layer->ValidateTensorShapesFromInputs(); + layer->ValidateTensorShapesFromInputs(shapeInferenceMethod); } } |