pytorch/torch/_higher_order_ops
2023-10-08 02:01:59 +00:00
..
__init__.py [HigherOrderOp] expose torch.cond (#110293) 2023-10-07 20:39:52 +00:00
cond.py [HigherOrderOp] expose torch.cond (#110293) 2023-10-07 20:39:52 +00:00
out_dtype.py [HigherOrderOp] fallthrough some keys by default. (#110478) 2023-10-05 16:25:42 +00:00
triton_kernel_wrap.py Add a side table for triton kernels to avoid using itertools.partial (#110633) 2023-10-08 02:01:59 +00:00
utils.py Higher order operator util for raising if inputs require grads (#106078) 2023-08-01 00:13:13 +00:00
wrap.py torch.compile + selective activation checkpointing (#105489) 2023-09-21 16:24:11 +00:00