Skip to content
HighVulnerability

CVE-2021-41206: TensorFlow operations missing tensor shape validation allow crashes

Identifier
CVE-2021-41206
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 operations in affected versions lack validation of the shapes of their tensor arguments. Depending on the API, this can cause undefined behavior, segfaults, or CHECK-fail crashes, and in some scenarios reads and writes to heap-populated arrays. The issues were found internally through tooling while testing GPU op determinism, and no reproducers exist.

Mitigation

Fixes will be included in TensorFlow 2.7.0. The commits will also be cherry-picked onto TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4, which are affected and still in supported range.