/python/dgl/nn/tensorflow/softmax.py

"""tf modules for graph related softmax."""
# pylint: disable= unused-import
from ..functional import edge_softmax