pytorch/torch/_subclasses
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
_fake_tensor_utils.py
fake_impls.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
fake_tensor.py Revert "Switch to using Python nested int (#141166)" 2024-11-22 23:54:36 +00:00
fake_utils.py check fake/real mismatches during real tensor prop (#137747) 2024-11-04 23:39:48 +00:00
functional_tensor.py [BE] Make maybe_aliasing_or_mutating proper tag (#131990) 2024-11-24 00:12:49 +00:00
meta_utils.py type annotations for meta_utils (#140203) 2024-11-13 20:07:47 +00:00
schema_check_mode.py