pytorch/torch/_prims
2025-06-16 07:38:34 +00:00
..
__init__.py [dynamic shapes] unbacked safe unsqueeze (#154087) 2025-05-30 01:41:57 +00:00
context.py remove allow-untyped-defs from context.py (#155622) 2025-06-16 07:38:34 +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