Skip to content
LowVulnerability

CVE-2021-29600: TensorFlow TFLite OneHot operator division by zero from crafted model

Identifier
CVE-2021-29600
Published
Record updated
View JSON
Affected
  • tensorflow < 2.1.4, fixed in 2.1.4, released June 15, 2021, 31 days after disclosure
  • tensorflow >= 2.2.0, < 2.2.3, fixed in 2.2.3, released June 15, 2021, 31 days after disclosure
  • tensorflow >= 2.3.0, < 2.3.3, fixed in 2.3.3, released June 15, 2021, 31 days after disclosure
  • and 9 more
Known exploitation
Not listed in the CISA Known Exploited Vulnerabilities catalog at the last check.
EPSS
0.2%

Summary

TensorFlow's implementation of the TFLite `OneHot` operator contains a division by zero error in one_hot.cc. An attacker can craft a model in which at least one dimension of `indices` is 0, which sets `prefix_dim_size` to 0 and triggers the flaw.

Mitigation

The fix will be included in TensorFlow 2.5.0. The commit will also be cherrypicked onto TensorFlow 2.4.2, 2.3.3, 2.2.3 and 2.1.4, which are affected and still in supported range.