https://github.com/deepmind/sonnet
Tip revision: 6cc140ece3d5c153ad7baaf4c95c9c9f295d01ad authored by Henning Becker on 02 April 2024, 16:38:14 UTC
Make Sonnet use CudnnRNNV3
Make Sonnet use CudnnRNNV3
Tip revision: 6cc140e
| File | Mode | Size |
|---|---|---|
| nets | ||
| src | ||
| BUILD | -rw-r--r-- | 2.4 KB |
| __init__.py | -rw-r--r-- | 5.0 KB |
| distribute.py | -rw-r--r-- | 1.1 KB |
| functional.py | -rw-r--r-- | 1.8 KB |
| initializers.py | -rw-r--r-- | 1.3 KB |
| mixed_precision.py | -rw-r--r-- | 990 bytes |
| optimizers.py | -rw-r--r-- | 1.0 KB |
| pad.py | -rw-r--r-- | 1009 bytes |
| regularizers.py | -rw-r--r-- | 971 bytes |
