Operators for Message-Passing#

Simple Neighborhood Aggregator (SAGEConv)#

`operators.agg_simple_n2n_fwd`	Computes the forward pass for simple aggregation (agg_simple) using node features in an
`operators.agg_simple_n2n_bwd`	Computes the backward pass for a simple aggregation (agg_simple) using node features in an
`operators.agg_simple_e2n_fwd`	Computes the forward pass for simple aggregation (agg_simple) using edge features in an
`operators.agg_simple_e2n_bwd`	Computes the backward pass for a simple aggregation (agg_simple) using edge features in an
`operators.agg_simple_n2n_e2n_fwd`	Computes the forward pass for both a simple aggregation (agg_simple) using node features in an
`operators.agg_simple_n2n_e2n_bwd`	Computes the backward pass for both a simple aggregation (agg_simple) using node features in an
`operators.agg_concat_n2n_fwd`	Computes the forward pass for simple aggregation using node features in an
`operators.agg_concat_n2n_bwd`	Computes the backward pass for a simple aggregation using node features in an
`operators.agg_concat_e2n_fwd`	Computes the forward pass for simple aggregation using edge features in an
`operators.agg_concat_e2n_bwd`	Computes the backward pass for a simple aggregation using edge features in an
`operators.agg_concat_n2n_e2n_fwd`	Computes the forward pass for both a simple aggregation using node features in an
`operators.agg_concat_n2n_e2n_bwd`	Computes the backward pass for both a simple aggregation (agg_simple) using node features in an

Weighted Neighborhood Aggregation#

`operators.agg_weighted_n2n_fwd`	Computes the forward pass for a weighted aggregation (agg_weighted) using node features in an
`operators.agg_weighted_n2n_bwd`	Computes the backward pass for a weighted aggregation (agg_weighted) using node features in an
`operators.agg_concat_weighted_n2n_fwd`	Computes the forward pass for a weighted aggregation (agg_weighted) using node features in an
`operators.agg_concat_weighted_n2n_bwd`	Computes the backward pass for a weighted aggregation (agg_weighted) using node features in an

Heterogenous Aggregator using Basis Decomposition (RGCNConv)#

`operators.agg_hg_basis_n2n_post_fwd`	Computes the forward pass for node-to-node full-graph RGCN-like basis regularized
`operators.agg_hg_basis_n2n_post_bwd`	Computes the backward pass for node-to-node full-graph RGCN-like basis regularized

Graph Attention (GATConv/GATv2Conv)#

`operators.mha_gat_n2n_fwd_bf16_fp32`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_fwd_fp16_fp32`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_fwd_fp32_fp32`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_bwd_bf16_bf16_bf16_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_bwd_bf16_bf16_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_bwd_bf16_fp32_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_bwd_fp16_fp16_fp16_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_bwd_fp16_fp16_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_bwd_fp16_fp32_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_bwd_fp32_fp32_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_fwd_bf16_fp32`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_fwd_fp16_fp32`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_fwd_fp32_fp32`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_bwd_bf16_bf16_bf16_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_bwd_bf16_bf16_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_bwd_bf16_fp32_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_bwd_fp16_fp16_fp16_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_bwd_fp16_fp16_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_bwd_fp16_fp32_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_n2n_efeat_bwd_fp32_fp32_fp32_fp32`	Computes the backward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_v2_n2n_fwd`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_v2_n2n_bwd`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_v2_n2n_efeat_fwd`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_v2_n2n_efeat_bwd`	Computes the backward pass for a multi-head attention layer (GAT-like)

Transformer-like Graph Attention (TransformerConv)#

`operators.mha_gat_v2_n2n_fwd`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_v2_n2n_bwd`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_v2_n2n_efeat_fwd`	Computes the forward pass for a multi-head attention layer (GAT-like)
`operators.mha_gat_v2_n2n_efeat_bwd`	Computes the backward pass for a multi-head attention layer (GAT-like)

Directional Message-Passing (DMPNN)#

`operators.agg_dmpnn_e2e_fwd`	Computes the forward pass for D-MPNN-like full graph aggregation
`operators.agg_dmpnn_e2e_bwd`	Computes the backward pass for D-MPNN-like full graph aggregation

Update Edges: Concatenation or Sum of Edge and Node Features#

`operators.update_efeat_e2e_concat_fwd`	Computes the forward pass for an operation which concatenates
`operators.update_efeat_e2e_concat_bwd`	Computes the backward pass for an operation which concatenates
`operators.update_efeat_e2e_sum_fwd`	Computes the forward pass for an operation which sums
`operators.update_efeat_e2e_sum_bwd`	Computes the backward pass for an operation which sums
`operators.update_efeat_e2e_concat_fwd`	Computes the forward pass for an operation which concatenates
`operators.update_efeat_e2e_concat_bwd`	Computes the backward pass for an operation which concatenates
`operators.update_efeat_e2e_sum_fwd`	Computes the forward pass for an operation which sums
`operators.update_efeat_e2e_sum_bwd`	Computes the backward pass for an operation which sums