These should be equivalent per the previous formula but breaks xla (#62329)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62329
Reviewed By: ejguan
Differential Revision: D29961527
Pulled By: albanD
fbshipit-source-id: 46e46726591f4c0c8faf6ec0d7136a2d4ca976ea