George Hotz 81b73f97a3 Optiimzation (#355) ... * constant folding into kernels * that opt worth it? * fix mypy * ast one kernel * save 2 lines in conv kernel * debug print kernel count * cl debugging * early realize inputs * refactor Device		2022-07-04 08:58:57 -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	make contiguous ops yellow	2022-07-02 17:54:04 -07:00
test_conv.py	a little more readable	2022-06-27 08:54:04 -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	test lazy also, make TestMNIST faster	2022-07-03 15:19:19 -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	Optiimzation (#355)	2022-07-04 08:58:57 -07:00
test_ops.py	add test for slice_one	2022-07-03 12:14:20 -07:00
test_optim.py	Test split (#231)	2021-01-01 09:19:03 -05:00
test_shapetracker.py	simple shapetracker tests	2022-07-02 16:02:15 -07:00
test_tensor.py	skip broken tests, no float64 allowed	2022-06-11 17:12:04 -07:00
test_train.py	the big memory gradient didn't even need to be computed	2022-06-16 11:41:29 -07:00