3 people f190ca446d Detach (#123) ...

* Detach

* Torch.detach reuses the buffer in the

* Fix test

* wakey wakey GitHub Actions

Co-authored-by: holonomicjl <[email protected]>

2020-11-19 19:03:42 -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

Detach (#123)

2020-11-19 19:03:42 -08:00

utils.py

move gradcheck to extra, clean up unbroadcast

2020-11-16 08:03:31 -08:00