pytorch/torch/_refs/nn
Ryan Spring 2df1da09e1 Add Elementwise unary ops 4 references (#78216)
Add reference implementations for `nan_to_num, positive, sigmoid, signbit, tanhshink`
Add prims for `minimum_value(dtype)` and `maximum_value(dtype)`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78216
Approved by: https://github.com/mruberry
2022-05-27 21:55:34 +00:00
..
functional Add Elementwise unary ops 4 references (#78216) 2022-05-27 21:55:34 +00:00
__init__.py Add logit ref; allow non-refs to be called in refs. 2022-05-21 02:35:14 +00:00