Handle non-persistent buffers and lifted tensors - Githubissues

justinchuby / torch-onnx

Prototype of the next torch exporter

MIT License

1 stars 1 forks source link

Handle non-persistent buffers and lifted tensors #66

Closed justinchuby closed 1 week ago

justinchuby commented 1 week ago

Reference https://github.com/pytorch/pytorch/blob/00f675bb4c2ec02bb5ffecfc75571026e220701c/torch/onnx/_internal/io_adapter.py#L584-L638

Fix https://github.com/justinchuby/torch-onnx/issues/60