Function get_lowered_module_name

exir/delegate.py:151–171

Adds the given lowered_module into the given root module and returns the name of the module added.

(
        root: torch.nn.Module,
        # pyre-ignore: Undefined or invalid type [11]: Annotation `LoweredBackendModule` is not defined as a type.
        lowered_module: LOWERED_BACKEND_MODULE_TYPE,  # noqa
    )

        return type(obj).__name__ == LOWERED_BACKEND_MODULE_TYPE

    def get_lowered_module_name(
        root: torch.nn.Module,
        # pyre-ignore: Undefined or invalid type [11]: Annotation `LoweredBackendModule` is not defined as a type.
        lowered_module: LOWERED_BACKEND_MODULE_TYPE,  # noqa
    ) -> str:
        """
        Adds the given lowered_module into the given root module and returns the
        name of the module added.
        """
        # Find a qualifying name for the lowered submodule
        qualname = None
        i = 0
        while True:
            qualname = f"lowered_module_{i}"
            if not hasattr(root, qualname):
                break
            i += 1
        assert qualname is not None

        root.add_module(qualname, lowered_module)
        return qualname