pytorch/torch/_decomp
Tugsbayasgalan Manlaibaatar 11c786dcb5 [BE] Make maybe_aliasing_or_mutating proper tag (#131990)
For better tracking, we need to make maybe aliasing/mutating ops with proper tag. We need to special case native_batch_norm because it is not a CIA but has a wrong schema. I guess native_batch_norm will be removed at some point, so until then we just keep it around.

D60347117
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131990
Approved by: https://github.com/bdhirsh
2024-11-24 00:12:49 +00:00
..
__init__.py [BE] Make maybe_aliasing_or_mutating proper tag (#131990) 2024-11-24 00:12:49 +00:00
decompositions_for_jvp.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
decompositions_for_rng.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
decompositions.py Added aten.bernoulli.p and aten.bernoulli.default decompositions (#139141) 2024-11-20 19:52:57 +00:00