Function _create_partitions

exir/backend/backend_api.py:534–558

(
    tagged_graph_module: torch.fx.GraphModule,
    partition_result: PartitionResult,
    owning_program: ExportedProgram,
    is_submodule: bool = False,
)

def _create_partitions(
    tagged_graph_module: torch.fx.GraphModule,
    partition_result: PartitionResult,
    owning_program: ExportedProgram,
    is_submodule: bool = False,
) -> Dict[str, List[torch.fx.Node]]:
    backend_id_to_call_submodules = _create_partitions_in_graph_module(
        tagged_graph_module, partition_result, owning_program, is_submodule
    )

    # Recursively partition and lower for submodules
    for _, submod, _ in get_control_flow_submodules(tagged_graph_module):
        nested_backend_id_to_call_submodules = _create_partitions(
            submod, partition_result, owning_program, is_submodule=True
        )
        for (
            backend_id,
            nested_submodules,
        ) in nested_backend_id_to_call_submodules.items():
            if backend_id not in backend_id_to_call_submodules:
                backend_id_to_call_submodules[backend_id] = nested_submodules
            else:
                backend_id_to_call_submodules[backend_id].extend(nested_submodules)

    return backend_id_to_call_submodules


def lower_all_submodules_to_backend(