Reference implementations for rsqrt and native_layer_norm (#79413)
This PR adds references for:
- `torch.rsqrt`
- `torch.native_layer_norm`
- `torch.nn.functional.layer_norm`
`native_layer_norm` had a different number of dimensions if the input was 0-sized. I fixed that.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79413
Approved by: https://github.com/mruberry, https://github.com/Chillee