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);
     }
 }