William Wen a8195f257e [custom_op] use new python custom ops API on prims ops (#124665) ... Also ads a non-decorator version of `custom_op`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/124665 Approved by: https://github.com/zou3519		2024-05-22 17:48:33 +00:00
..
__init__.py	[custom_op] use new python custom ops API on prims ops (#124665)	2024-05-22 17:48:33 +00:00
context.py
debug_prims.py
executor.py
rng_prims.py	[custom_op] use new python custom ops API on prims ops (#124665)	2024-05-22 17:48:33 +00:00