Implement __tvm_ffi_object__ protocol.
(self)
| 148 | self._tensor = tensor |
| 149 | |
| 150 | def __tvm_ffi_object__(self) -> tvm_ffi.Tensor: |
| 151 | """Implement __tvm_ffi_object__ protocol.""" |
| 152 | return self._tensor |
| 153 | |
| 154 | data: npt.NDArray[Any] = np.zeros((10, 8, 4, 2), dtype="int32") |
| 155 | if not hasattr(data, "__dlpack__"): |