geohotstan 8c0d0a122c Add return_indices to max_pool (#9506) ... * wow argmax is so good * 1 less line * clean up and better variable names * is this torch thing right...? * add more tests * slap a TODO on it * clean ups * prettier looking code and fix ceil mode test * add return types and some docs * ok that was a bad example since indices == value, just no example		2025-03-19 15:25:37 -04:00
..
.gitignore	torch backend works for ResNet-18 (#9200)	2025-02-22 22:16:23 +08:00
backend.py	Add return_indices to max_pool (#9506)	2025-03-19 15:25:37 -04:00
backend2.py	torch and numpy dtype interop [pr] (#9224)	2025-02-24 18:26:49 +08:00
example.py	run some tests from torch [pr] (#9252)	2025-02-26 15:42:22 +08:00
test.py	add masked_select to tensor.py (#9468)	2025-03-17 16:05:36 -04:00
test_compile.py	start torch.compile support (#9279)	2025-02-27 10:29:51 +08:00
test_inplace.py	torch: support in-place operations on views (#9371)	2025-03-10 23:29:00 +08:00
test_multigpu.py	torch backend multigpu - add devices and tests (#9414)	2025-03-12 11:33:11 +08:00
torch_tests.py	rig up torch's testing framework [pr] (#9254)	2025-02-26 18:46:22 +08:00
wrapped_tensor.cpp	torch backend multigpu - add devices and tests (#9414)	2025-03-12 11:33:11 +08:00