geohotstanandGitHub 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

backend2.py

torch and numpy dtype interop [pr] (#9224)

2025-02-24 18:26:49 +08:00

backend.py

Add return_indices to max_pool (#9506)

2025-03-19 15:25:37 -04:00

example.py

run some tests from torch [pr] (#9252)

2025-02-26 15:42:22 +08: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

test.py

add masked_select to tensor.py (#9468)

2025-03-17 16:05:36 -04: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