pytorch
2b8c5605 - extend view function to accomodate node/number args, generate constraints for scalar addition and lt (#80823)

Commit
2 years ago
extend view function to accomodate node/number args, generate constraints for scalar addition and lt (#80823) - View function can contain arguments that are nodes or numbers. Before, we were assuming all the arguments are numbers which is not general. We modify the code to handle this case. Some of the other inference rules will also be updated accordingly. - Generate scalar addition constraints of the form input = output - Generate lt constraints which are the same as those for gt. Pull Request resolved: https://github.com/pytorch/pytorch/pull/80823 Approved by: https://github.com/jamesr66a
Author
Committer
Parents
Loading