George Hotz 6261a0639b ShapeTracker (#328) ...

* start shapetracker

* that late reshape is crushing our hopes

* simple failure

* DumbShapeTracker passes tests

* improve st tests

* stacked view tracker works

* flip works

* tests pass

* shapetracker works

* use ShapeTracker in ops_gpu

* a couple lines

* fix 0 shape

* less lines

* use shapetracker for new_shape in ops.py

* simpler still

* padding with a ZeroView

* gamed it a little

2022-06-14 16:08:22 -07:00

..

efficientnet

add ViT test and car

2022-06-05 17:12:43 -07:00

__init__.py

All devices are equal! (#196)

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

test_conv.py

refactor to pass conv args into llops

2022-06-11 23:08:46 -07:00

test_efficientnet.py

test_onnx works with enet also

2022-06-11 14:30:26 -07:00

test_gc.py

Test split (#231)

2021-01-01 09:19:03 -05:00

test_llops.py

move ops to ops.py

2022-06-11 15:58:56 -07:00

test_mnist.py

sgd threestep graph is so pretty

2022-06-06 09:45:37 -07:00

test_net_speed.py

Test split (#231)

2021-01-01 09:19:03 -05:00

test_nn.py

fix bn test, affine is True

2022-01-15 19:52:15 -08:00

test_onnx.py

add padding to conv2d function, other minor things

2022-06-11 22:29:42 -07:00

test_ops.py

Expand Operator (#327)

2022-06-12 12:31:48 -07:00

test_optim.py

Test split (#231)

2021-01-01 09:19:03 -05:00

test_shapetracker.py

ShapeTracker (#328)

2022-06-14 16:08:22 -07:00

test_tensor.py

skip broken tests, no float64 allowed

2022-06-11 17:12:04 -07:00

test_train.py

bs 4 -> 2 in training test

2022-01-15 21:34:21 -08:00