|
__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 |
|
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 |