Index

A

add_conv_relational_bias() (in module opengt.layer.nodeformer_layer) add_graph_token() (in module opengt.encoder.graphormer_encoder) aggregate() (opengt.layer.gatedgcn_layer.GatedGCNLayer method)

APPNP (class in opengt.layer.appnp_layer) ASTEdgeEncoder (class in opengt.encoder.ast_encoder) ASTNodeEncoder (class in opengt.encoder.ast_encoder)

B

BGA (class in opengt.network.bga_model)

BGALayer (class in opengt.layer.bga_layer) BigBird (class in opengt.network.big_bird)

C

CoBFormer (class in opengt.network.cobformer) compute_posenc_stats() (in module opengt.transform.posenc_stats)

concat_node_encoders() (in module opengt.encoder.composed_encoders) custom_eigh() (in module opengt.transform.posenc_stats) CustomGNN (class in opengt.network.custom_gnn)

D

DeGTA (class in opengt.network.degta_model)

DeGTAConv (class in opengt.layer.degta_layer) DIFFormer (class in opengt.network.difformer)

E

eigvec_normalizer() (in module opengt.transform.posenc_stats) ElstaticSENodeEncoder (class in opengt.encoder.kernel_pos_encoder) EquivStableLapPENodeEncoder (class in opengt.encoder.equivstable_laplace_pos_encoder) EREdgeEncoder (class in opengt.encoder.ER_edge_encoder) ERNodeEncoder (class in opengt.encoder.ER_node_encoder) ETransformer (class in opengt.layer.ETransformer)

ExpanderEdgeFixer (class in opengt.encoder.exp_edge_fixer) ExphormerAttention (class in opengt.layer.Exphormer) ExphormerFullLayer (class in opengt.layer.Exphormer) extra_repr() (opengt.layer.gps_layer.GPSLayer method) (opengt.layer.multi_model_layer.MultiLayer method) (opengt.layer.multi_model_layer.SingleLayer method)

F

FeatureEncoder (class in opengt.encoder.feature_encoder) (class in opengt.network.multi_model)

FFN (class in opengt.layer.bga_layer) full_edge_index() (in module opengt.encoder.rrwp_encoder)

G

GatedGCNGraphGymLayer (class in opengt.layer.gatedgcn_layer) GatedGCNLayer (class in opengt.layer.gatedgcn_layer) get_electrostatic_function_encoding() (in module opengt.transform.posenc_stats) get_heat_kernels() (in module opengt.transform.posenc_stats) get_heat_kernels_diag() (in module opengt.transform.posenc_stats) get_lap_decomp_stats() (in module opengt.transform.posenc_stats) get_rw_landing_probs() (in module opengt.transform.posenc_stats) GINDeepSigns (class in opengt.encoder.signnet_pos_encoder) GINEConvESLapPE (class in opengt.layer.gine_conv_layer) GINEConvGraphGymLayer (class in opengt.layer.gine_conv_layer) GINEConvLayer (class in opengt.layer.gine_conv_layer)

GNNInductiveEdgeHead (class in opengt.head.inductive_edge) GNNInductiveNodeHead (class in opengt.head.inductive_node) GPSLayer (class in opengt.layer.gps_layer) GPSModel (class in opengt.network.gps_model) GraphMLPMixer (class in opengt.network.mlp_mixer) GraphormerHead (class in opengt.head.graphormer_graph) GraphormerLayer (class in opengt.layer.graphormer_layer) GraphormerModel (class in opengt.network.graphormer) Graphtransformer (class in opengt.network.graphtransformer) GritTransformer (class in opengt.network.grit_model) GritTransformerLayer (class in opengt.layer.grit_layer)

H

HKdiagSENodeEncoder (class in opengt.encoder.kernel_pos_encoder)

I

InferLinksHead (class in opengt.head.infer_links)

K

kernelized_gumbel_softmax() (in module opengt.layer.nodeformer_layer)

kernelized_softmax() (in module opengt.layer.nodeformer_layer) KernelPENodeEncoder (class in opengt.encoder.kernel_pos_encoder)

L

LapPENodeEncoder (class in opengt.encoder.laplace_pos_encoder) LinearEdgeEncoder (class in opengt.encoder.linear_edge_encoder)

LinearNodeEncoder (class in opengt.encoder.linear_node_encoder) load_dataset_master() (in module opengt.loader.master_loader)

M

MaskedGINDeepSigns (class in opengt.encoder.signnet_pos_encoder) message() (opengt.layer.gatedgcn_layer.GatedGCNLayer method) (opengt.layer.gine_conv_layer.GINEConvESLapPE method) MLPMixer (class in opengt.layer.mlp_mixer) MLPMixerGraphHead (class in opengt.head.mlp_mixer_graph) module opengt.config.data_preprocess_config opengt.config.defaults_config opengt.config.gt_config opengt.encoder.ast_encoder opengt.encoder.composed_encoders opengt.encoder.equivstable_laplace_pos_encoder opengt.encoder.ER_edge_encoder opengt.encoder.ER_node_encoder opengt.encoder.exp_edge_fixer opengt.encoder.feature_encoder opengt.encoder.graphormer_encoder opengt.encoder.kernel_pos_encoder opengt.encoder.laplace_pos_encoder opengt.encoder.linear_edge_encoder opengt.encoder.linear_node_encoder opengt.encoder.ppa_encoder opengt.encoder.rrwp_encoder opengt.encoder.signnet_pos_encoder opengt.encoder.sine_encoder opengt.encoder.spd_encoder opengt.encoder.type_dict_encoder opengt.encoder.voc_superpixels_encoder opengt.encoder.wl_encoder opengt.head.graphormer_graph opengt.head.inductive_edge opengt.head.inductive_node opengt.head.infer_links opengt.head.mlp_mixer_graph opengt.head.ogb_code_graph opengt.head.san_graph opengt.layer.appnp_layer opengt.layer.bga_layer opengt.layer.degta_layer opengt.layer.ETransformer opengt.layer.Exphormer opengt.layer.gatedgcn_layer opengt.layer.gine_conv_layer opengt.layer.gps_layer opengt.layer.graphormer_layer opengt.layer.grit_layer opengt.layer.mlp_mixer opengt.layer.multi_model_layer opengt.layer.nodeformer_layer opengt.layer.other_attn_layer opengt.layer.patch_encoder opengt.layer.san2_layer opengt.layer.san_layer opengt.layer.spec_layer opengt.layer.trans_conv_layer opengt.loader.master_loader opengt.network.bga_model opengt.network.big_bird opengt.network.cobformer opengt.network.custom_gnn opengt.network.degta_model opengt.network.difformer opengt.network.gps_model opengt.network.graphormer opengt.network.graphtransformer opengt.network.grit_model opengt.network.mlp_mixer opengt.network.multi_model opengt.network.nodeformer opengt.network.performer opengt.network.san_transformer opengt.network.sgformer opengt.network.specformer opengt.network.trans_conv opengt.transform.posenc_stats opengt.transform.task_preprocessing

MultiHeadAttention (class in opengt.layer.bga_layer) MultiHeadAttention2Layer (class in opengt.layer.san2_layer) MultiHeadAttentionLayer (class in opengt.layer.san_layer) MultiHeadAttentionLayerGraphormerSparse (class in opengt.layer.other_attn_layer) MultiHeadAttentionLayerGritSparse (class in opengt.layer.grit_layer) MultiHeadAttentionLayerSANSparse (class in opengt.layer.other_attn_layer) MultiLayer (class in opengt.layer.multi_model_layer) MultiModel (class in opengt.network.multi_model)

N

NodeFormer (class in opengt.network.nodeformer)

NodeFormerConv (class in opengt.layer.nodeformer_layer)

O

OGBCodeGraphHead (class in opengt.head.ogb_code_graph) opengt.config.data_preprocess_config module opengt.config.defaults_config module opengt.config.gt_config module opengt.encoder.ast_encoder module opengt.encoder.composed_encoders module opengt.encoder.equivstable_laplace_pos_encoder module opengt.encoder.ER_edge_encoder module opengt.encoder.ER_node_encoder module opengt.encoder.exp_edge_fixer module opengt.encoder.feature_encoder module opengt.encoder.graphormer_encoder module opengt.encoder.kernel_pos_encoder module opengt.encoder.laplace_pos_encoder module opengt.encoder.linear_edge_encoder module opengt.encoder.linear_node_encoder module opengt.encoder.ppa_encoder module opengt.encoder.rrwp_encoder module opengt.encoder.signnet_pos_encoder module opengt.encoder.sine_encoder module opengt.encoder.spd_encoder module opengt.encoder.type_dict_encoder module opengt.encoder.voc_superpixels_encoder module opengt.encoder.wl_encoder module opengt.head.graphormer_graph module opengt.head.inductive_edge module opengt.head.inductive_node module opengt.head.infer_links module opengt.head.mlp_mixer_graph module opengt.head.ogb_code_graph module opengt.head.san_graph module opengt.layer.appnp_layer module opengt.layer.bga_layer module opengt.layer.degta_layer module opengt.layer.ETransformer module opengt.layer.Exphormer module

opengt.layer.gatedgcn_layer module opengt.layer.gine_conv_layer module opengt.layer.gps_layer module opengt.layer.graphormer_layer module opengt.layer.grit_layer module opengt.layer.mlp_mixer module opengt.layer.multi_model_layer module opengt.layer.nodeformer_layer module opengt.layer.other_attn_layer module opengt.layer.patch_encoder module opengt.layer.san2_layer module opengt.layer.san_layer module opengt.layer.spec_layer module opengt.layer.trans_conv_layer module opengt.loader.master_loader module opengt.network.bga_model module opengt.network.big_bird module opengt.network.cobformer module opengt.network.custom_gnn module opengt.network.degta_model module opengt.network.difformer module opengt.network.gps_model module opengt.network.graphormer module opengt.network.graphtransformer module opengt.network.grit_model module opengt.network.mlp_mixer module opengt.network.multi_model module opengt.network.nodeformer module opengt.network.performer module opengt.network.san_transformer module opengt.network.sgformer module opengt.network.specformer module opengt.network.trans_conv module opengt.transform.posenc_stats module opengt.transform.task_preprocessing module overwrite_defaults_cfg() (in module opengt.config.defaults_config)

P

PadToFullGraphEdgeEncoder (class in opengt.encoder.rrwp_encoder) PatchEncoder (class in opengt.layer.patch_encoder) Performer (class in opengt.network.performer) PPAEdgeEncoder (class in opengt.encoder.ppa_encoder)

PPANodeEncoder (class in opengt.encoder.ppa_encoder) pyg_softmax() (in module opengt.layer.grit_layer) (in module opengt.layer.other_attn_layer) (in module opengt.layer.san2_layer)

R

reset_parameters() (opengt.layer.gine_conv_layer.GINEConvESLapPE method) RRWPLinearEdgeEncoder (class in opengt.encoder.rrwp_encoder)

RRWPLinearEdgeMaskedEncoder (class in opengt.encoder.rrwp_encoder) RRWPLinearNodeEncoder (class in opengt.encoder.rrwp_encoder) RWSENodeEncoder (class in opengt.encoder.kernel_pos_encoder)

S

SAN2Layer (class in opengt.layer.san2_layer) SANGraphHead (class in opengt.head.san_graph) SANLayer (class in opengt.layer.san_layer) SANTransformer (class in opengt.network.san_transformer) ScaledDotProductAttention (class in opengt.layer.bga_layer) set_cfg_gt() (in module opengt.config.gt_config) set_cfg_preprocess() (in module opengt.config.data_preprocess_config)

SGFormer (class in opengt.network.sgformer) SignNetNodeEncoder (class in opengt.encoder.signnet_pos_encoder) SineEncoder (class in opengt.encoder.sine_encoder) SingleLayer (class in opengt.layer.multi_model_layer) SingleModel (class in opengt.network.multi_model) SPDEdgeEncoder (class in opengt.encoder.spd_encoder) SpecFormer (class in opengt.network.specformer) SpecLayer (class in opengt.layer.spec_layer)

T

task_specific_preprocessing() (in module opengt.transform.task_preprocessing) TransConv (class in opengt.network.trans_conv)

TransConvLayer (class in opengt.layer.trans_conv_layer) TypeDictEdgeEncoder (class in opengt.encoder.type_dict_encoder) TypeDictNodeEncoder (class in opengt.encoder.type_dict_encoder)

U

update() (opengt.layer.gatedgcn_layer.GatedGCNLayer method)

V

VOCEdgeEncoder (class in opengt.encoder.voc_superpixels_encoder)

VOCNodeEncoder (class in opengt.encoder.voc_superpixels_encoder)

W

WLSENodeEncoder (class in opengt.encoder.wl_encoder)