add comment for rope operator

This commit is contained in:
xiaonans 2024-02-04 10:40:25 +08:00
parent 9a3c0f11f6
commit ae9f61de5a
1 changed files with 8 additions and 0 deletions

View File

@ -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;