pytorch/torch/_prims
2025-05-30 01:41:57 +00:00
..
__init__.py [dynamic shapes] unbacked safe unsqueeze (#154087) 2025-05-30 01:41:57 +00:00
context.py PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102) 2025-01-18 20:47:12 +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