diff options
Diffstat (limited to 'include/armnn/INetwork.hpp')
| -rw-r--r-- | include/armnn/INetwork.hpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index c4c67b70ae..aaca13bba9 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -17,7 +17,6 @@ namespace armnn { - /// @brief An input connection slot for a layer. /// The input slot can be connected to an output slot of the preceding layer in the graph. /// Only one connection to the input slot is allowed. @@ -48,6 +47,10 @@ public: virtual int Connect(IInputSlot& destination) = 0; virtual void Disconnect(IInputSlot& slot) = 0; + virtual unsigned int CalculateIndexOnOwner() const = 0; + + virtual LayerGuid GetOwningLayerGuid() const = 0; + protected: /// Not user deletable. ~IOutputSlot() {} @@ -361,6 +364,8 @@ public: /// @ return - Interface for configuring the layer. virtual IConnectableLayer* AddGatherLayer(const char* name = nullptr) = 0; + virtual void Accept(ILayerVisitor& visitor) const = 0; + protected: ~INetwork() {} }; |