Does cuDNN's 2**31 elements limit apply to TF 2.5.0?

General Discussion
#1

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:

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

1 Like