Method module

torch/export/exported_program.py:349–359

Returns a self contained GraphModule with all the parameters/buffers inlined.

(self, *, flat: bool = True)

        return string

    def module(self, *, flat: bool = True) -> torch.nn.Module:
        """
        Returns a self contained GraphModule with all the parameters/buffers inlined.
        """
        from torch._export.unflatten import unflatten
        from ._unlift import _unlift_exported_program_lifted_states

        if flat:
            return _unlift_exported_program_lifted_states(self)
        else:
            return unflatten(self)

    @_disable_prexisiting_fake_mode
    def run_decompositions(