Method set_var_remap

python/tvm/relax/expr_functor.py:1462–1473

Remap a var to a new var in use-site. Parameters ---------- vid : Id The vid of the old var. var : Var The new var.

(self, vid: Id, var: Var)

        return _ffi_api.PyExprMutatorVisitExprPostOrder(self._outer(), expr)  # type: ignore

    def set_var_remap(self, vid: Id, var: Var) -> None:
        """Remap a var to a new var in use-site.

        Parameters
        ----------
        vid : Id
            The vid of the old var.
        var : Var
            The new var.
        """
        # Using self._outer() to ref _PyExprMutator
        return _ffi_api.PyExprMutatorSetVarRemap(self._outer(), vid, var)  # type: ignore

    def get_var_remap(self, vid: Id) -> Var:
        """Remap a var to a new var in use-site.