torch_geometric.profile .. currentmodule:: torch_geometric.profile .. autosummary:: :nosignatures: {% for cls in torch_geometric.profile.classes %} {{ cls }} {% endfor %} .. automodule:: torch_geometric.profile :members: :undoc-members: