<filename>python/tvm/contrib/msir/torch/transform/transform.py
import tvm
from tvm.contrib.msir.core.utils import MSIR_TARGET
from tvm.contrib.msir.core.transform import partition_utils
from tvm.contrib.msir.core import utils as msir_utils


def partition_for_torch(mod,
    params = None,
    msir_name = "msir_torch",
    is_train = False,
    need_prune = False,
    need_quantize = False,
    need_collect = False):
    return partition_utils.target_partition(mod, \
        params = params,
        msir_name = msir_name,
        target = MSIR_TARGET.TORCH,
        is_train = is_train,
        need_prune = need_prune,
        need_quantize = need_quantize,
        need_collect = need_collect
    )
<filename>python/tvm/contrib/msir/torch/transform/transform.py
import tvm
from tvm.contrib.msir.core.utils import MSIR_TARGET
from tvm.contrib.msir.core.transform import partition_utils
from tvm.contrib.msir.core import utils as msir_utils


def partition_for_torch(mod,
    params = None,
    msir_name = "msir_torch",
    is_train = False,
    need_prune = False,
    need_quantize = False,
    need_collect = False):
    return partition_utils.target_partition(mod, \
        params = params,
        msir_name = msir_name,
        target = MSIR_TARGET.TORCH,
        is_train = is_train,
        need_prune = need_prune,
        need_quantize = need_quantize,
        need_collect = need_collect
    )