pytorch

1d9f445b - moved logit to use torch ops instead of refs + added a couple more decompositions

moved logit to use torch ops instead of refs + added a couple more decompositions Pull Request resolved: https://github.com/pytorch/pytorch/pull/78984 Approved by: https://github.com/ezyang