Skip to content
MediumVulnerability

CVE-2021-37640: TensorFlow SparseReshape division by zero via tf.raw_ops

Identifier
CVE-2021-37640
Published
Record updated
View JSON
Affected
  • tensorflow < 2.3.4, fixed in 2.3.4, released August 10, 2021, 2 days before disclosure
  • tensorflow >= 2.4.0, < 2.4.3, fixed in 2.4.3, released August 10, 2021, 2 days before disclosure
  • tensorflow = 2.5.0, fixed in 2.5.1, released August 10, 2021, 2 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-37640 affects TensorFlow versions before the fix. The `tf.raw_ops.SparseReshape` implementation triggers an integral division by 0 exception because the reshaping functor divides by the target shape's dimensions without checking that the input or target shape has a non-zero number of elements.

Mitigation

Patched in GitHub commit 4923de56ec94fff7770df259ab7f2288a74feb41. The fix will be included in TensorFlow 2.6.0, and the commit will also be cherry-picked to TensorFlow 2.5.1.