tinygrad/tinygrad
Mufeed VH 0bbf66627c
Define ProfileOp class once (#131)
* define `ProfileOp` class once

* clean `ProfileOp` class

* removed `else: pass`
2020-11-24 19:39:13 -08:00
..
__init__.py factor out a couple nn ops 2020-10-29 08:01:12 -07:00
nn.py promote swish to a tensor ops 2020-11-10 15:48:11 -08:00
ops.py unbroadcast GPU template 2020-11-16 08:16:36 -08:00
opsgpu.py unbroadcasting works on the GPU 2020-11-16 09:16:55 -08:00
optim.py tensor implementation for rmsprop and adam (#121) 2020-11-16 15:07:49 -08:00
tensor.py Define ProfileOp class once (#131) 2020-11-24 19:39:13 -08:00
utils.py move gradcheck to extra, clean up unbroadcast 2020-11-16 08:03:31 -08:00