Segfault in TensorFlow if `tf.histogram_fixed_width` is called with NaN values

Summary

TensorFlow is an open source platform for machine learning. Prior to versions 2.9.0, 2.8.1, 2.7.2, and 2.6.4, the implementation of `tf.histogram_fixed_width` is vulnerable to a crash when the values array contain `Not a Number` (`NaN`) elements. The implementation assumes that all floating point operations are defined and then converts a floating point result to an integer index. If `values` contains `NaN` then the result of the division is still `NaN` and the cast to `int32` would result in a crash. This only occurs on the CPU implementation. Versions 2.9.0, 2.8.1, 2.7.2, and 2.6.4 contain a patch for this issue.

Timeline

1. May 20, 2022, 11:20 PM UTCCVE published
2. Apr 22, 2025, 3:46 PM UTCCISA SSVC assessment

   Exploitation: poc · Automatable: no · Technical impact: partial

3. Apr 22, 2025, 5:57 PM UTCRecord last updated

Affected versions

• tensorflow tensorflow

  fixed in 2.6.4; From 2.7.0rc0 · fixed in 2.7.2; From 2.8.0rc0 · fixed in 2.8.1; From 2.9.0rc0 · fixed in 2.9.0

• tensorflowPyPI

  fixed in 2.6.4

• tensorflowPyPI

  From 2.7.0 · fixed in 2.7.2

• tensorflowPyPI

  From 2.8.0 · fixed in 2.8.1

• tensorflow-cpuPyPI

  fixed in 2.6.4

• tensorflow-cpuPyPI

  From 2.7.0 · fixed in 2.7.2

• tensorflow-cpuPyPI

  From 2.8.0 · fixed in 2.8.1

• tensorflow-gpuPyPI

  fixed in 2.6.4

• tensorflow-gpuPyPI

  From 2.7.0 · fixed in 2.7.2

• tensorflow-gpuPyPI

  From 2.8.0 · fixed in 2.8.1

• tensorflow-cpuPyPI

  fixed in 2.6.4; From 2.7.0 · fixed in 2.7.2; From 2.8.0 · fixed in 2.8.1

• tensorflow-gpuPyPI

  fixed in 2.6.4; From 2.7.0 · fixed in 2.7.2; From 2.8.0 · fixed in 2.8.1

• tensorflowPyPI

  fixed in 2.6.4; From 2.7.0 · fixed in 2.7.2; From 2.8.0 · fixed in 2.8.1

References 9