diff options
Diffstat (limited to 'src/armnn/Graph.hpp')
| -rw-r--r-- | src/armnn/Graph.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp index 731ae1e5aa..d5fbeafed0 100644 --- a/src/armnn/Graph.hpp +++ b/src/armnn/Graph.hpp @@ -203,6 +203,10 @@ public: void SubstituteSubgraph(SubgraphView& subgraph, IConnectableLayer* substituteLayer); void SubstituteSubgraph(SubgraphView& subgraph, const SubgraphView& substituteSubgraph); + /// For each ConstantLayer in Graph, ensures TensorInfo is set on all output slots. + /// LayerValidationException thrown if no TensorInfo is set. + void VerifyConstantLayerSetTensorInfo() const; + void InferTensorInfos(); void AttachObservable(IGraphObservable* const observable, GraphEvent notifyOnEvent) { |