pytorch

2bfae07a - Enable `dim=None` for `torch.mean` (#81286)

Enable `dim=None` for `torch.mean` (#81286) Part of #79525 This will require coordination with XLA before merging, just like #79881 Pull Request resolved: https://github.com/pytorch/pytorch/pull/81286 Approved by: https://github.com/albanD