diff --git a/docs/api/attention.md b/docs/api/attention.md index 8630c563..c3dddcfd 100644 --- a/docs/api/attention.md +++ b/docs/api/attention.md @@ -53,12 +53,24 @@ what runs when you call `op(...)`. heading_level: 3 members: ["__init__", "forward"] +::: tileops.attention.MLADenseBwdOp + options: + show_root_heading: true + heading_level: 3 + members: ["__init__", "forward"] + ::: tileops.attention.MLAPagedFwdOp options: show_root_heading: true heading_level: 3 members: ["__init__", "forward"] +::: tileops.attention.MLAVarlenBwdOp + options: + show_root_heading: true + heading_level: 3 + members: ["__init__", "forward"] + ::: tileops.attention.MLAVarlenFwdOp options: show_root_heading: true