Skip to content

Commit

Permalink
Update operators.rst (#4339)
Browse files Browse the repository at this point in the history
  • Loading branch information
raybellwaves authored May 20, 2024
1 parent 30c5edb commit 0056302
Showing 1 changed file with 20 additions and 4 deletions.
24 changes: 20 additions & 4 deletions docs/cugraph/source/api_docs/cugraph-ops/python/operators.rst
Original file line number Diff line number Diff line change
Expand Up @@ -47,10 +47,26 @@ Graph Attention (GATConv/GATv2Conv)
.. autosummary::
:toctree: ../../api/ops

operators.mha_gat_n2n_fwd
operators.mha_gat_n2n_bwd
operators.mha_gat_n2n_efeat_fwd
operators.mha_gat_n2n_efeat_bwd
operators.mha_gat_n2n_fwd_bf16_fp32
operators.mha_gat_n2n_fwd_fp16_fp32
operators.mha_gat_n2n_fwd_fp32_fp32
operators.mha_gat_n2n_bwd_bf16_bf16_bf16_fp32
operators.mha_gat_n2n_bwd_bf16_bf16_fp32_fp32
operators.mha_gat_n2n_bwd_bf16_fp32_fp32_fp32
operators.mha_gat_n2n_bwd_fp16_fp16_fp16_fp32
operators.mha_gat_n2n_bwd_fp16_fp16_fp32_fp32
operators.mha_gat_n2n_bwd_fp16_fp32_fp32_fp32
operators.mha_gat_n2n_bwd_fp32_fp32_fp32_fp32
operators.mha_gat_n2n_efeat_fwd_bf16_fp32
operators.mha_gat_n2n_efeat_fwd_fp16_fp32
operators.mha_gat_n2n_efeat_fwd_fp32_fp32
operators.mha_gat_n2n_efeat_bwd_bf16_bf16_bf16_fp32
operators.mha_gat_n2n_efeat_bwd_bf16_bf16_fp32_fp32
operators.mha_gat_n2n_efeat_bwd_bf16_fp32_fp32_fp32
operators.mha_gat_n2n_efeat_bwd_fp16_fp16_fp16_fp32
operators.mha_gat_n2n_efeat_bwd_fp16_fp16_fp32_fp32
operators.mha_gat_n2n_efeat_bwd_fp16_fp32_fp32_fp32
operators.mha_gat_n2n_efeat_bwd_fp32_fp32_fp32_fp32

operators.mha_gat_v2_n2n_fwd
operators.mha_gat_v2_n2n_bwd
Expand Down

0 comments on commit 0056302

Please sign in to comment.