dgl.nn (TensorFlow)

Conv Layers

dgl.nn.tensorflow.conv.GraphConv
dgl.nn.tensorflow.conv.RelGraphConv
dgl.nn.tensorflow.conv.GATConv
dgl.nn.tensorflow.conv.SAGEConv
dgl.nn.tensorflow.conv.ChebConv
dgl.nn.tensorflow.conv.SGConv
dgl.nn.tensorflow.conv.APPNPConv
dgl.nn.tensorflow.conv.GINConv

Global Pooling Layers

dgl.nn.tensorflow.glob.SumPooling
dgl.nn.tensorflow.glob.AvgPooling
dgl.nn.tensorflow.glob.MaxPooling
dgl.nn.tensorflow.glob.SortPooling
dgl.nn.tensorflow.glob.GlobalAttentionPooling

Heterogeneous Learning Modules

dgl.nn.tensorflow.glob.HeteroGraphConv