pytorch/torch/_higher_order_ops
2024-12-17 09:42:36 +00:00
..
__init__.py [foreach_map] Initial foreach map HOP impl for inference (#142098) 2024-12-11 21:32:11 +00:00
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 Remove some unused type ignores (round 1) (#142325) 2024-12-09 18:23:46 +00:00
executorch_call_delegate.py
flex_attention.py [FlexAttention] Fix broken eager tracing (#143344) 2024-12-17 09:42:36 +00:00
foreach_map.py [foreach_map] Initial foreach map HOP impl for inference (#142098) 2024-12-11 21:32:11 +00:00
hints_wrap.py
invoke_subgraph.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
map.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
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 Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
utils.py [while_loop] data-dependent op in body_fn (#142031) 2024-12-10 21:54:28 +00:00
while_loop.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
wrap.py