pytorch/torch/_prims
2025-06-11 19:44:18 +00:00
..
__init__.py [dynamic shapes] unbacked safe unsqueeze (#154087) 2025-05-30 01:41:57 +00:00
context.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
debug_prims.py
executor.py remove allow-untyped-defs from torch/_prims/executor.py (#144233) 2025-01-07 19:40:40 +00:00
rng_prims.py Support torch.compile rng selective activation checkpointing with cudagraph (#146878) 2025-02-28 00:47:03 +00:00