pytorch
bc1fef96 - Reference implementations for rsqrt and native_layer_norm (#79413)

Commit
2 years ago
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
Author
Committer
Parents
Loading