CVE-2021-29552

critical
Published 2021-05-21 · Modified 2023-12-06
CVSS v3
CVSS:3.1/AV:L/AC:H/PR:L/UI:N/S:U/C:N/I:N/A:L
CVSS v2
VIR risk
9.5

Description

TensorFlow is an end-to-end open source platform for machine learning. An attacker can cause a denial of service by controlling the values of `num_segments` tensor argument for `UnsortedSegmentJoin`. This is because the implementation(https://github.com/tensorflow/tensorflow/blob/a2a607db15c7cd01d754d37e5448d72a13491bdb/tensorflow/core/kernels/unsorted_segment_join_op.cc#L92-L93) assumes that the `num_segments` tensor is a valid scalar. Since the tensor is empty the `CHECK` involved in `.scalar<T>()()` that checks that the number of elements is exactly 1 will be invalidated and this would result in process termination. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2, TensorFlow 2.3.3, TensorFlow 2.2.3 and TensorFlow 2.1.4, as these are also affected and still in supported range.

Predictions

Exploit likelihood
20%
Patch ETA

Heuristic predictions, AS-IS, for prioritization only.

Mitigations

vendor Authored 2026-05-27

Vendor advisory: debian — https://security-tracker.debian.org/tracker/CVE-2021-29552

OS impact

OSVersionStatusFixed in
arch archfixed2.5.0-1
debian debianforkyfixed0
debian debiansidfixed0

Package impact

EcosystemPackageVulnerableFixed
python PyPItensorflow<2.1.42.1.4
python PyPItensorflow>=2.2.0,<2.2.32.2.3
python PyPItensorflow>=2.3.0,<2.3.32.3.3
python PyPItensorflow>=2.4.0,<2.4.22.4.2
python PyPItensorflow-cpu<2.1.42.1.4
python PyPItensorflow-cpu>=2.2.0,<2.2.32.2.3
python PyPItensorflow-cpu>=2.3.0,<2.3.32.3.3
python PyPItensorflow-cpu>=2.4.0,<2.4.22.4.2
python PyPItensorflow-gpu<2.1.42.1.4
python PyPItensorflow-gpu>=2.2.0,<2.2.32.2.3
python PyPItensorflow-gpu>=2.3.0,<2.3.32.3.3
python PyPItensorflow-gpu>=2.4.0,<2.4.22.4.2
python PyPItensorflow<704866eabe03a9aeda044ec91a8d0c83fc1ebdbe||>=2.4.0,<2.4.3704866eabe03a9aeda044ec91a8d0c83fc1ebdbe
python PyPItensorflow-cpu<704866eabe03a9aeda044ec91a8d0c83fc1ebdbe||>=2.4.0,<2.4.3704866eabe03a9aeda044ec91a8d0c83fc1ebdbe
python PyPItensorflow-gpu<704866eabe03a9aeda044ec91a8d0c83fc1ebdbe||>=2.4.0,<2.4.3704866eabe03a9aeda044ec91a8d0c83fc1ebdbe

References

Verify integrity in audit chain (admin only). AS-IS.