pytorch

9705fb03 - Add support for a couple ops

Add support for a couple ops Pull Request resolved: https://github.com/pytorch/pytorch/pull/79581 Approved by: https://github.com/Chillee