tvm_ffi.from_dlpack

Contents

tvm_ffi.from_dlpack#

tvm_ffi.from_dlpack(ext_tensor, *, require_alignment=0, require_contiguous=False)#

Convert an external tensor to an Tensor.

Parameters:

  • ext_tensor (object) – The external tensor to convert.

  • require_alignment (int) – The minimum required alignment to check for the tensor.

  • require_contiguous (bool) – Whether to check for contiguous memory.

Returns:

tensor – The converted tensor.

Return type:

tvm_ffi.Tensor

Contents