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