Function _wrapper

python/tvm/contrib/dlpack.py:39–44

(*args)

    assert callable(tvm_func)

    def _wrapper(*args):
        args = tuple(
            tvm.runtime.from_dlpack(to_dlpack_func(arg)) if isinstance(arg, tensor_type) else arg
            for arg in args
        )
        return tvm_func(*args)

    return _wrapper