From ae9f61de5ae1fc26c9257b1e4ca71eca4528ca1d Mon Sep 17 00:00:00 2001
From: xiaonans <sxnzxz@gmail.com>
Date: Sun, 4 Feb 2024 10:40:25 +0800
Subject: [PATCH] add comment for rope operator

---
 include/operators/rope.h | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/include/operators/rope.h b/include/operators/rope.h
index 2eb312fe..b21adb24 100644
--- a/include/operators/rope.h
+++ b/include/operators/rope.h
@@ -4,6 +4,14 @@
 namespace infini {
 class RoPEObj : public OperatorObj {
   public:
+    /**
+     * @brief Construct a new RotaryEmbedding object.
+     *
+     * @param graph The computation graph that this operator belongs to.
+     * @param pos The positon id of the query.
+     * @param input The input tensor.
+     * @param output The output tensor.
+     */
     RoPEObj(GraphObj *graph, Tensor pos, Tensor input, Tensor output);
     OP_CLONE(RoPEObj);
     optional<vector<Shape>> inferShape(const TensorVec &inputs) override;