Sample code:

from tvm.contrib.dlpack import to_pytorch_func
m_pytorch = to_pytorch_func(lib\['default'\](ctx))

torch_output = torch.empty(1, 40, 120, 256)
m_pytorch(input1, input2, input3, torch_output)





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/dlpack-error-attributeerror-module-has-no-function-tvm-main/8029/2)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/ad2e83da728c653b5432bc5389e391b829e751ac5fe712eb187dfd3880e626e6).

Reply via email to