def get_graph_info(graph):
    """
    Wrapper fn that calculates number of trainable vars in an MTF graph & prints all dim_names to file
    TODO: how to get un-trainable dim-names too, batch etc.

    :param graph: Mesh-Tensorflow graph
    :return: None
    """
    get_n_trainable_vars(graph)
    print_dim_names(graph)
Wrapper fn that calculates number of trainable vars in an MTF graph & prints all dim_names to file
TODO: how to get un-trainable dim-names too, batch etc.

:param graph: Mesh-Tensorflow graph
:return: None
utils.py