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> inferShape(const TensorVec &inputs) override;