tinygrad/extra/torch_backend
George Hotz 8ee3a37524
shrink/pad use (new_shape, offset) (#16405)
* shrink uses offset and shape

* pad does too

* fix
2026-05-27 15:13:08 -07:00
..
.gitignore torch backend works for ResNet-18 (#9200) 2025-02-22 22:16:23 +08:00
backend.py shrink/pad use (new_shape, offset) (#16405) 2026-05-27 15:13:08 -07:00
backend2.py torch and numpy dtype interop [pr] (#9224) 2025-02-24 18:26:49 +08:00
example.py put the ranges on store instead of after (#15759) 2026-04-16 19:06:40 +08:00
test.py Torch backend aten::cat.out fix (#16121) 2026-05-11 16:28:16 -07:00
test_compile.py start torch.compile support (#9279) 2025-02-27 10:29:51 +08:00
test_inplace.py remove allow_shape_mismatch in Tensor.replace (#14536) 2026-02-04 12:38:18 -05:00
test_kernel_fusion.py requires_grad cannot be None (#16212) 2026-05-15 02:01:04 -04:00
test_multigpu.py move frontend dir to nn [pr] (#12470) 2025-10-07 10:42:22 +08:00
torch_tests.py rig up torch's testing framework [pr] (#9254) 2025-02-26 18:46:22 +08:00
wrapped_tensor.cpp remove allow_shape_mismatch in Tensor.replace (#14536) 2026-02-04 12:38:18 -05:00