コード例 #1
0
ファイル: sharding.py プロジェクト: vishalbelsare/pytorch
 def traverse_graph(graph):
     results = set()
     for datapipe, sub_graph in graph.items():
         results.add(datapipe)
         sub_items = traverse_graph(sub_graph)
         for item in sub_items:
             results.add(item)
     return results
コード例 #2
0
ファイル: graph_settings.py プロジェクト: paolodedios/pytorch
def get_all_graph_pipes(graph):
    results = set()
    for datapipe, sub_graph in graph.items():
        results.add(datapipe)
        sub_items = get_all_graph_pipes(sub_graph)
        for item in sub_items:
            results.add(item)
    return results