Hzfengsy opened a new pull request, #14463: URL: https://github.com/apache/tvm/pull/14463
The current `arange` op in PyTorch is converted as constant at frontend, however, it does not work when the length is dynamic. This PR adds `arange` op to relax, and use different rules when legalization: - If the length is fixed, we will convert it to constant. - If the length is dynamic, we use `topi.arange` to compute it. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
