Method _convert_tvm_to_pytorch

python/tvm/relax/base_py_module.py:420–426

Convert TVM Tensors to PyTorch tensors using DLPack.

(
        self, tvm_tensors: Any | list[Any]
    )

            ) from error

    def _convert_tvm_to_pytorch(
        self, tvm_tensors: Any | list[Any]
    ) -> Union["torch.Tensor", list["torch.Tensor"]]:
        """Convert TVM Tensors to PyTorch tensors using DLPack."""
        if isinstance(tvm_tensors, list | tuple):
            return [self._convert_single_tvm_to_pytorch(tensor) for tensor in tvm_tensors]
        return self._convert_single_tvm_to_pytorch(tvm_tensors)

    def _convert_single_tvm_to_pytorch(self, tvm_tensor: Any) -> "torch.Tensor":
        """Convert a single TVM Tensor to PyTorch tensor using faster DLPack converter."""