Method from_expr

python/tvm/ir/module.py:199–218

Construct a module from a standalone expression. Parameters ---------- expr: RelaxExpr The starting expression global_funcs: Optional[dict] Map of global vars to function definitions Returns ------- mod: Module

(expr, functions=None)

    @staticmethod
    def from_expr(expr, functions=None):
        """Construct a module from a standalone expression.

        Parameters
        ----------
        expr: RelaxExpr
            The starting expression

        global_funcs: Optional[dict]
            Map of global vars to function definitions

        Returns
        -------
        mod: Module
            A module containing the passed definitions,
            where expr is set as the entry point
            (wrapped in a function if necessary)
        """
        funcs = functions if functions is not None else {}
        return _ffi_api.Module_FromExpr(expr, funcs)

    def get_attr(self, attr_key):
        """Get the IRModule attribute.