Joel Schlosser 869d629c0f Forward / backward NJT support for several activation functions (#140736) ... Several activation functions were unimplemented due to missing `pointwise` tags. This PR adds them and corresponding backwards implementations. Pull Request resolved: https://github.com/pytorch/pytorch/pull/140736 Approved by: https://github.com/Skylion007, https://github.com/cpuhrsch ghstack dependencies: #141500		2024-11-26 21:19:58 +00:00
..
__init__.py
nested_tensor.py	Revert "Switch to using Python nested int (#141166)"	2024-11-22 23:54:36 +00:00
ops.py	Forward / backward NJT support for several activation functions (#140736)	2024-11-26 21:19:58 +00:00
sdpa.py	Misc. non-contig NJT fixes (#140160)	2024-11-09 01:18:26 +00:00