graph G {
    compound="true"
    rankdir="TB"
    bgcolor="white"
    fontname="Tahoma"

    node [
        fixedsize="false"
        fontname="Tahoma"
        color="white"
        fillcolor="deepskyblue2"
        fontcolor="black"
        shape="box"
        style="filled"
        penwidth="1.0"
    ]
    edge [
        fontname="Arial"
        color="#00688b"
        fontcolor="black"
        fontsize="12"
        arrowsize="0.5"
        penwidth="1.0"
    ]




    "[mesh_tensorflow/transformer/transformer_layers.py]" -- "[mesh_tensorflow/transformer/attention.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[mesh_tensorflow/transformer/moe.py]" -- "[mesh_tensorflow/transformer/attention.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[mesh_tensorflow/transformer/transformer.py]" -- "[mesh_tensorflow/transformer/attention.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[mesh_tensorflow/transformer/transformer.py]" -- "[mesh_tensorflow/transformer/moe.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[mesh_tensorflow/transformer/transformer_layers.py]" -- "[mesh_tensorflow/transformer/moe.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[mesh_tensorflow/transformer/transformer_layers.py]" -- "[mesh_tensorflow/transformer/transformer.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[mesh_tensorflow/transformer/utils.py]" -- "[mesh_tensorflow/optimize.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[mesh_tensorflow/ops.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[mesh_tensorflow/simd_mesh_impl.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[mesh_tensorflow/simd_mesh_impl.py]" -- "[mesh_tensorflow/ops.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[mesh_tensorflow/optimize.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/transformer/utils.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/__init__.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/__init__.py]" -- "[mesh_tensorflow/optimize.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/__init__.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/__init__.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/__init__.py]" -- "[mesh_tensorflow/optimize.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/__init__.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/__init__.py]" -- "[mesh_tensorflow/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/api.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/api.py]" -- "[mesh_tensorflow/optimize.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/api.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/api.py]" -- "[mesh_tensorflow/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/api.py]" -- "[mesh_tensorflow/auto_mtf/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/graph_interface.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/graph_interface.py]" -- "[mesh_tensorflow/optimize.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/graph_interface.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/graph_interface.py]" -- "[mesh_tensorflow/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/graph_interface.py]" -- "[mesh_tensorflow/auto_mtf/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/graph_interface.py]" -- "[mesh_tensorflow/auto_mtf/api.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/layout_optimizer.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/layout_optimizer.py]" -- "[mesh_tensorflow/optimize.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/layout_optimizer.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/layout_optimizer.py]" -- "[mesh_tensorflow/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/layout_optimizer.py]" -- "[mesh_tensorflow/auto_mtf/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/layout_optimizer.py]" -- "[mesh_tensorflow/auto_mtf/api.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/layout_optimizer.py]" -- "[mesh_tensorflow/auto_mtf/graph_interface.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/memory_estimator.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/memory_estimator.py]" -- "[mesh_tensorflow/optimize.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/memory_estimator.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/memory_estimator.py]" -- "[mesh_tensorflow/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/memory_estimator.py]" -- "[mesh_tensorflow/auto_mtf/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/memory_estimator.py]" -- "[mesh_tensorflow/auto_mtf/api.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/memory_estimator.py]" -- "[mesh_tensorflow/auto_mtf/graph_interface.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/memory_estimator.py]" -- "[mesh_tensorflow/auto_mtf/layout_optimizer.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/print_cp_model_solution.py]" -- "[mesh_tensorflow/auto_mtf/valid_layouts.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/print_cp_model_solution.py]" -- "[mesh_tensorflow/optimize.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/print_cp_model_solution.py]" -- "[mesh_tensorflow/transformer/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/print_cp_model_solution.py]" -- "[mesh_tensorflow/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[mesh_tensorflow/auto_mtf/print_cp_model_solution.py]" -- "[mesh_tensorflow/auto_mtf/__init__.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];

}