Hi there,
TensorFlow 2.5.0 is based on cuDNN 8.x, which no longer has the 2 billion elements per tensor limitation of cuDNN 7.x:
https://docs.nvidia.com/deeplearning/cudnn/release-notes/rel_8.html 9
I would like to know: does TensorFlow have a limitation of its own, independently of cuDNN, limiting the number of elements in a tensor?
Thank you,
OD