torch_geometric.profile torch_geometric.profile {% for cls in torch_geometric.profile.classes %} {{ cls }} {% endfor %} torch_geometric.profile