aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/frontend/ILayer.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph/frontend/ILayer.h')
-rw-r--r--arm_compute/graph/frontend/ILayer.h22
1 files changed, 22 insertions, 0 deletions
diff --git a/arm_compute/graph/frontend/ILayer.h b/arm_compute/graph/frontend/ILayer.h
index 5add8ab440..f7caaea47d 100644
--- a/arm_compute/graph/frontend/ILayer.h
+++ b/arm_compute/graph/frontend/ILayer.h
@@ -46,6 +46,28 @@ public:
* @return ID of the created node.
*/
virtual NodeID create_layer(IStream &s) = 0;
+ /** Sets the name of the layer
+ *
+ * @param[in] name Name of the layer
+ *
+ * @return The layer object
+ */
+ ILayer &set_name(std::string name)
+ {
+ _name = name;
+ return *this;
+ }
+ /** Layer name accessor
+ *
+ * @return Returns the name of the layer
+ */
+ const std::string &name() const
+ {
+ return _name;
+ }
+
+private:
+ std::string _name = {};
};
} // namespace frontend
} // namespace graph