pytorch

b651148f - remove prims::square (#78627)

remove prims::square (#78627) because it is just `x * x` Pull Request resolved: https://github.com/pytorch/pytorch/pull/78627 Approved by: https://github.com/mruberry