pytorch/torch/_higher_order_ops
2025-01-03 19:01:07 +00:00
..
__init__.py
associative_scan.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
auto_functionalize.py
cond.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
effects.py
executorch_call_delegate.py
flex_attention.py [FlexAttention] Fix broken eager tracing (#143344) 2024-12-17 09:42:36 +00:00
foreach_map.py
hints_wrap.py
invoke_subgraph.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
map.py
out_dtype.py
prim_hop_base.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
run_const_graph.py
scan.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
strict_mode.py
torchbind.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
triton_kernel_wrap.py [user triton] Raise an exception when encountering nested @triton.autotune decorators or @triton.heuristics (#143519) 2024-12-20 06:38:45 +00:00
utils.py [cond] Change Autograd for cond (#142518) 2024-12-19 20:09:42 +00:00
while_loop.py [while_loop][aot] auto-unspecialize int input and output to unbacked symints (#143105) 2025-01-03 19:01:07 +00:00
wrap.py