nn-tensorflow.rst

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