diff options
Diffstat (limited to 'src/armnn/Graph.hpp')
| -rw-r--r-- | src/armnn/Graph.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp index d5fbeafed0..e2321bb0e4 100644 --- a/src/armnn/Graph.hpp +++ b/src/armnn/Graph.hpp @@ -268,6 +268,11 @@ private: std::map<const GraphEvent, std::list<IGraphObservable*>> m_Views; ShapeInferenceMethod m_ShapeInferenceMethod; + + // Throws exception due to a layer input not being connected to an output slot. + /// Also verifies weights and bias are set for FullyConnected layers. + void ConstructErrorMessageForUnconnectedInputs(Layer* const layer, + unsigned int slotIndex); }; /// Common base class for layers in the graph. |