Skip to content
MediumVulnerability

CVE-2021-41202: TensorFlow integer overflow in tf.range kernel output size calculation

Identifier
CVE-2021-41202
Published
Record updated
View JSON
Affected
  • tensorflow >= 2.6.0, < 2.6.1, fixed in 2.6.1, released November 1, 2021, 5 days before disclosure
  • tensorflow >= 2.5.0, < 2.5.2, fixed in 2.5.2, released November 1, 2021, 5 days before disclosure
  • tensorflow < 2.4.4, fixed in 2.4.4, released November 1, 2021, 5 days before disclosure
  • and 6 more
Known exploitation
Not listed in the CISA Known Exploited Vulnerabilities catalog at the last check.
EPSS
0.2%

Summary

TensorFlow's `tf.range` kernel computes the output size with a conditional of type `int64 = condition ? int64 : double`. C++ implicit conversion casts both branches to `double` and truncates the result before assignment, causing overflows in affected versions.

Mitigation

The fix will be included in TensorFlow 2.7.0. The commit will also be cherry-picked to TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4.