CVE-2021-41195 - CVE House
Back to Database
Status published Medium CVE-2021-41195

Crash in `tf.math.segment_*` operations

Vulnerability Description

TensorFlow is an open source platform for machine learning. In affected versions the implementation of `tf.math.segment_*` operations results in a `CHECK`-fail related abort (and denial of service) if a segment id in `segment_ids` is large. This is similar to CVE-2021-29584 (and similar other reported vulnerabilities in TensorFlow, localized to specific APIs): the implementation (both on CPU and GPU) computes the output shape using `AddDim`. However, if the number of elements in the tensor overflows an `int64_t` value, `AddDim` results in a `CHECK` failure which provokes a `std::abort`. Instead, code should use `AddDimWithStatus`. The fix will be included in TensorFlow 2.7.0. We will also cherrypick this commit on TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4, as these are also affected and still in supported range.

Impact Analysis

Refer to official advisory for detailed impact metrics.

Remediation

Ensure systems are updated to the latest vendor-supplied patch levels.

THREAT MONITOR

Am I Vulnerable?

Launch our assessment wizard to check if your infrastructure is exposed to • CVE-2021-41195

Credits & Attribution

No credits recorded in the NVD database.

References

More from tensorflow

View All →
CVE-2024-3660
Arbitrary code injection vulnerability in Keras framework < 2.13
Unknown 0
CVE-2023-33976
TensorFlow segfault in array_ops.upper_bound
High 7.5
CVE-2023-27579
TensorFlow has Floating Point Exception in TFLite in conv kernel
High 7.5
CVE-2023-25801
TensorFlow has double free in Fractional(Max/Avg)Pool
High 8
CVE-2023-25676
TensorFlow has null dereference on ParallelConcat with XLA
High 7.5

Affected Vendor

Affected Software

tensorflow
Vulnerable Versions:
>= 2.6.0, < 2.6.1, >= 2.5.0, < 2.5.2, < 2.4.4

Timeline

Official Publish: November 5th, 2021
Last Modified: August 4th, 2024
Added to House: July 21st, 2026

CVSS Vectors

V3: CVSS:3.1/AV:L/AC:L/PR:L/UI:N/S:U/C:N/I:N/A:H

Weaknesses (CWE)

CWE-190

MITRE ATT&CK TTPs

T1190 Exploit Public-Facing Application
Initial Access
T1068 Exploitation for Privilege Escalation
Privilege Escalation
T1055 Process Injection
Defense Evasion
T1105 Ingress Tool Transfer
Command and Control
T1499 Endpoint Denial of Service
Impact