Revision 6cc140ece3d5c153ad7baaf4c95c9c9f295d01ad authored by Henning Becker on 02 April 2024, 16:38:14 UTC, committed by Copybara-Service on 02 April 2024, 16:39:00 UTC
CudnnRNN and CudnnRNNV2 are not compatible with cuDNN 9+, so this change makes Sonnet use CudnnRNNV3 instead. Note that this raises the minimum supported cuDNN version to 8.1 (which is below 8.9 - the minimum supported cuDNN version in Tensorflow anyway). PiperOrigin-RevId: 621206198
1 parent 26b0518
File | Mode | Size |
---|---|---|
.github | ||
docs | ||
examples | ||
sonnet | ||
CONTRIBUTING.md | -rw-r--r-- | 1.7 KB |
LICENSE | -rw-r--r-- | 11.1 KB |
MANIFEST.in | -rw-r--r-- | 34 bytes |
README.md | -rw-r--r-- | 11.4 KB |
WORKSPACE | -rw-r--r-- | 27 bytes |
readthedocs.yml | -rw-r--r-- | 357 bytes |
requirements-test.txt | -rw-r--r-- | 46 bytes |
requirements-tf.txt | -rw-r--r-- | 53 bytes |
requirements.txt | -rw-r--r-- | 74 bytes |
setup.py | -rw-r--r-- | 2.0 KB |
test.sh | -rwxr-xr-x | 1.8 KB |
Computing file changes ...