George Hotz 02655c07d5 break maxpool2d on GPU

2020-12-29 13:05:57 -05:00

..

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

test_gc.py

add transpose, start on transformer

2020-12-27 16:59:12 -05:00

test_mnist.py

fixup training loop

2020-12-27 18:35:56 -05:00

test_net_speed.py

add transpose, start on transformer

2020-12-27 16:59:12 -05:00

test_nn.py

add transpose, start on transformer

2020-12-27 16:59:12 -05:00

test_ops.py

break maxpool2d on GPU

2020-12-29 13:05:57 -05:00

test_optim.py

add transpose, start on transformer

2020-12-27 16:59:12 -05:00

test_tensor.py

add transpose, start on transformer

2020-12-27 16:59:12 -05:00