Skip to content
MediumVulnerability

CVE-2021-41218: TensorFlow divide by zero in AllToAll shape inference

Identifier
CVE-2021-41218
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.1%

Summary

CVE-2021-41218 affects TensorFlow's shape inference code for `AllToAll`. When the `split_count` argument is 0, the code can execute a division by zero (CWE-369).

Mitigation

Fixed in TensorFlow 2.7.0. The commit is also cherry-picked to TensorFlow 2.6.1, 2.5.2 and 2.4.4.