Skip to content
HighVulnerability

CVE-2021-41226: TensorFlow heap out-of-bounds access in SparseBinCount

Identifier
CVE-2021-41226
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

CVE-2021-41226 affects TensorFlow versions where the `SparseBinCount` implementation performs a heap out-of-bounds access. The cause is missing validation between the elements of the `values` argument and the shape of the sparse output.

Mitigation

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