Method with_struct_info

python/tvm/relax/expr_functor.py:1524–1541

Create a new var with specified shape and type if the original var's shape or type does not match with the specified ones. Parameters ---------- var : Var The var to be updated. struct_info : StructInfo The struct info. Return

(self, var: Var, struct_info: StructInfo)

        return _ffi_api.PyExprMutatorLookupBinding(self._outer(), var)  # type: ignore

    def with_struct_info(self, var: Var, struct_info: StructInfo) -> Var:
        """Create a new var with specified shape and type if the original var's shape or type does
        not match with the specified ones.

        Parameters
        ----------
        var : Var
            The var to be updated.
        struct_info : StructInfo
            The struct info.

        Returns
        -------
        var : Var
            The var filled with shape and type.
        """
        # Using self._outer() to ref _PyExprMutator
        return _ffi_api.PyExprMutatorWithStructInfo(self._outer(), var, struct_info)  # type: ignore