Skip to content
LowVulnerability

CVE-2021-29578: TensorFlow heap buffer overflow in tf.raw_ops.FractionalAvgPoolGrad

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

The implementation of `tf.raw_ops.FractionalAvgPoolGrad` in TensorFlow contains a heap buffer overflow. The kernel in `fractional_avg_pool_op.cc` does not check that the pooling sequence arguments have enough elements for the shape of the `out_backprop` tensor.

Mitigation

The fix will be included in TensorFlow 2.5.0. The commit will also be cherry-picked to TensorFlow 2.4.2, 2.3.3, 2.2.3 and 2.1.4, which are affected and still in supported range.