CVE-2021-29560: TensorFlow Heap Buffer Overflow in tf.raw_ops.RaggedTensorToTensor
Summary
CVE-2021-29560: TensorFlow is an end-to-end open source platform for machine learning. Affects tensorflow-gpu across 5 version ranges.
>= 2.2.0, < 2.3.0rc0
< 2.2.0rc0
>= 2.3.0, < 2.3.4
>= 2.4.0, < 2.4.3
< a84358aa12f0b1518e606095ab9cfddbf597c121
Fixed in 4 releases; the minimum fixed version depends on the release line you run. How to fix ↓
Also tracked for
- tensorflow · Bitnami
- tensorflow · PyPI
- tensorflow · PyPI
- tensorflow-cpu · PyPI
The same CVE-2021-29560 is tracked separately for each ecosystem and vendor. Its fix information is the same; the package and affected versions differ.
Check if this vulnerability affects your application
Upload a manifest or SBOM and we'll show whether you're affected, plus every other risky package in the same scan.
No signup required · Results in about 60 seconds
Background
TensorFlow is an end-to-end open source platform for machine learning. It provides a comprehensive ecosystem of tools, libraries, and resources for building and deploying machine learning models. TensorFlow is widely used in various domains, including research, industry, and academia. The vulnerability described here affects the 'tensorflow-gpu' package, which is a PyPI package for TensorFlow with GPU support.
Vulnerability Detail
The vulnerability exists in the 'tf.raw_ops.RaggedTensorToTensor' function of TensorFlow. Due to a flaw in the implementation, an attacker can trigger a heap buffer overflow by controlling the shape of the input arguments. Specifically, when the 'parent_output_index' is shorter than 'row_split', a heap out-of-bounds access can occur. This can lead to arbitrary code execution or a denial-of-service condition. The vulnerability has been assigned the identifier CVE-2021-29560.
How To Fix
Latest published version: 2.12.0 as of 22 Oct 2026
pip install --upgrade "tensorflow-gpu==2.12.0"Upgrading to it may include breaking changes; the versions below are the minimum fixed release for this vulnerability.
If you can't move to the latest
The minimum fixed version depends on the release line you run.
-
< 2.2.0rc02.2.0rc0pip install --upgrade "tensorflow-gpu==2.2.0rc0" -
>= 2.2.0, < 2.3.0rc02.3.0rc0pip install --upgrade "tensorflow-gpu==2.3.0rc0" -
>= 2.3.0, < 2.3.42.3.4pip install --upgrade "tensorflow-gpu==2.3.4" -
>= 2.4.0, < 2.4.32.4.3pip install --upgrade "tensorflow-gpu==2.4.3"
Tip — Not sure which release line you're on? Vulert can find the fix for the version you are using. Scan free →
Workaround
There is no known workaround for this vulnerability. It is recommended to apply the fix by updating to the fixed versions of TensorFlow.
Conclusion
The vulnerability in TensorFlow's 'tf.raw_ops.RaggedTensorToTensor' function can be exploited to trigger a heap buffer overflow. To mitigate this vulnerability, users should update to the fixed versions of TensorFlow. For further support and assistance in vulnerability remediation, you can contact DevNack.com.
See every vulnerability recorded for tensorflow-gpu →
FAQ
TensorFlow is an open source platform for machine learning that provides tools and resources for building and deploying machine learning models.
The vulnerability can be exploited by an attacker to trigger a heap buffer overflow, potentially leading to arbitrary code execution or a denial-of-service condition.
The vulnerability affects TensorFlow versions 2.2.0rc0 and above, including TensorFlow 2.5.0.
No, there are no known workarounds for this vulnerability. It is recommended to update to the fixed versions of TensorFlow.
You can find more information about this vulnerability in the official TensorFlow advisory and the CVE entry.
References
Related vulnerabilities
Recent vulnerabilities recorded for tensorflow-gpu.
