Skip to content

Commit 0056302

Browse files
authored
1 parent 30c5edb commit 0056302

File tree

1 file changed

+20
-4
lines changed

1 file changed

+20
-4
lines changed

docs/cugraph/source/api_docs/cugraph-ops/python/operators.rst

Lines changed: 20 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,26 @@ Graph Attention (GATConv/GATv2Conv)
4747
.. autosummary::
4848
:toctree: ../../api/ops
4949

50-
operators.mha_gat_n2n_fwd
51-
operators.mha_gat_n2n_bwd
52-
operators.mha_gat_n2n_efeat_fwd
53-
operators.mha_gat_n2n_efeat_bwd
50+
operators.mha_gat_n2n_fwd_bf16_fp32
51+
operators.mha_gat_n2n_fwd_fp16_fp32
52+
operators.mha_gat_n2n_fwd_fp32_fp32
53+
operators.mha_gat_n2n_bwd_bf16_bf16_bf16_fp32
54+
operators.mha_gat_n2n_bwd_bf16_bf16_fp32_fp32
55+
operators.mha_gat_n2n_bwd_bf16_fp32_fp32_fp32
56+
operators.mha_gat_n2n_bwd_fp16_fp16_fp16_fp32
57+
operators.mha_gat_n2n_bwd_fp16_fp16_fp32_fp32
58+
operators.mha_gat_n2n_bwd_fp16_fp32_fp32_fp32
59+
operators.mha_gat_n2n_bwd_fp32_fp32_fp32_fp32
60+
operators.mha_gat_n2n_efeat_fwd_bf16_fp32
61+
operators.mha_gat_n2n_efeat_fwd_fp16_fp32
62+
operators.mha_gat_n2n_efeat_fwd_fp32_fp32
63+
operators.mha_gat_n2n_efeat_bwd_bf16_bf16_bf16_fp32
64+
operators.mha_gat_n2n_efeat_bwd_bf16_bf16_fp32_fp32
65+
operators.mha_gat_n2n_efeat_bwd_bf16_fp32_fp32_fp32
66+
operators.mha_gat_n2n_efeat_bwd_fp16_fp16_fp16_fp32
67+
operators.mha_gat_n2n_efeat_bwd_fp16_fp16_fp32_fp32
68+
operators.mha_gat_n2n_efeat_bwd_fp16_fp32_fp32_fp32
69+
operators.mha_gat_n2n_efeat_bwd_fp32_fp32_fp32_fp32
5470

5571
operators.mha_gat_v2_n2n_fwd
5672
operators.mha_gat_v2_n2n_bwd

0 commit comments

Comments
 (0)