diff options
Diffstat (limited to 'src/armnn/Graph.hpp')
| -rw-r--r-- | src/armnn/Graph.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp index d71149d069..0c34d35685 100644 --- a/src/armnn/Graph.hpp +++ b/src/armnn/Graph.hpp @@ -280,6 +280,8 @@ private: /// Also verifies weights and bias are set for FullyConnected layers. void ConstructErrorMessageForUnconnectedInputs(Layer* const layer, unsigned int slotIndex); + + friend class SubgraphView; }; /// Common base class for layers in the graph. |