George Hotz 49c6e6d472 Latest attempt to add image (#462) ... * add image * load + store + boring stuff: * image tests pass * thneed print GFLOPS * op conv test * more debugging * hack for multiview image * shapetracker creates less views * disable image tests * working better * ugh, lkey not key * print in DEBUG, and allow views * works * simple padding conv2d * use index for image * that was bad code * debug print * fix types * less lines * save lines		2023-01-12 17:36:30 -08:00
..
gemm	Simple chonker (#431)	2022-11-10 23:17:09 -08:00
augment.py	Reshape dataset from fetch_mnist (#390)	2022-09-24 21:16:29 -04:00
gradcheck.py	move gradcheck to extra, clean up unbroadcast	2020-11-16 08:03:31 -08:00
introspection.py	Simple version of the new GPU backend (#458)	2023-01-10 19:16:02 -08:00
onnx.py	move batchnorm to Tensor	2023-01-09 18:00:16 -08:00
thneed.py	Latest attempt to add image (#462)	2023-01-12 17:36:30 -08:00
training.py	prune graph	2022-07-17 15:38:43 -07:00
utils.py	move get_parameters to optim.py	2022-09-25 13:16:58 -04:00