Method _convert_pytorch_to_tvm

python/tvm/relax/base_py_module.py:362–371

Convert PyTorch tensors to TVM Tensors using DLPack.

(
        self, tensors: Any | list[Any] | tuple[Any, ...]
    )

        return dtype_mapping.get(str(tvm_dtype), torch.float32)

    def _convert_pytorch_to_tvm(
        self, tensors: Any | list[Any] | tuple[Any, ...]
    ) -> Tensor | list[Tensor]:
        """Convert PyTorch tensors to TVM Tensors using DLPack."""
        # pylint: disable=import-outside-toplevel
        import torch

        if isinstance(tensors, list | tuple):
            return [self._convert_single_pytorch_to_tvm(t) for t in tensors]
        return self._convert_single_pytorch_to_tvm(tensors)

    def _convert_single_pytorch_to_tvm(self, tensor: Any) -> Tensor:
        """Convert a single PyTorch tensor to TVM Tensor with faster DLPack converter."""