George Hotz 5d45c6e516 Fold reduce (#362) ...

* folding reduce

* fold through movementops

* fixup shapes

* was too aggressive

* i knew we needed that

* don't recompute reduce

* working

* fix openpilot compile

* prunegraph openpilot

* types and reduce_shape

* refactor

* cleanups

* neater

* 1009

* 1004

* clean up reduce for 998

2022-07-19 09:24:02 -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

graph_batchnorm.py

prune graph

2022-07-17 15:38:43 -07:00

test_conv.py

Fold reduce (#362)

2022-07-19 09:24:02 -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_mnist.py

prune graph

2022-07-17 15:38:43 -07:00

test_net_speed.py

more crap to remove without convs

2022-07-17 13:02:27 -07:00

test_nn.py

w/e, that's a later prob

2022-07-17 12:32:50 -07:00

test_onnx.py

prints show we can precompute from the outside

2022-07-08 10:59:20 -07:00

test_ops.py

Fold reduce (#362)

2022-07-19 09:24:02 -07:00

test_optim.py

Test split (#231)

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

test_shapetracker.py

SLICE -> PAD,SHRINK

2022-07-17 11:33:59 -07:00

test_tensor.py

skip broken tests, no float64 allowed

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

test_train.py

opencl not imported

2022-07-17 12:14:39 -07:00