Allow batch sizes > 65535 for inverse, solve, cholesky_solve and tria… (#21689)
Summary:
…ngular_solve
Changelog:
- Iterate over mini batches of 65535 matrices (maximum)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21689
Differential Revision: D15800254
Pulled By: soumith
fbshipit-source-id: c743ff13f1ba25d26874429d44e41a3c0ed21d6a