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