Register for the AI for Security Summit: Join Figma, Perplexity & Wiz

CVE-2021-29544
Python vulnerability analysis and mitigation

Overview

TensorFlow, an end-to-end open source platform for machine learning, was found to contain a vulnerability (CVE-2021-29544) that could trigger a denial of service via a CHECK-fail in tf.raw_ops.QuantizeAndDequantizeV4Grad. The vulnerability was discovered by Yakun Zhang and Ying Wang of Baidu X-Team and was disclosed on May 13, 2021. The affected versions include TensorFlow 2.4.0 and 2.4.1 (GitHub Advisory).

Technical details

The vulnerability occurs because the implementation does not validate the rank of the input_* tensors. When these tensors are passed to QuantizeAndDequantizePerChannelGradientImpl, the vec method requires the rank to be 1 and triggers a CHECK failure if this requirement is not met. The issue can be triggered by providing input tensors with incorrect ranks to the tf.raw_ops.QuantizeAndDequantizeV4Grad operation (GitHub Advisory).

Impact

An attacker can exploit this vulnerability to cause a denial of service condition in applications using the affected TensorFlow versions. The impact is considered Low severity according to the official advisory (GitHub Advisory).

Exploitability

The vulnerability can be triggered by providing specifically crafted input tensors to the tf.raw_ops.QuantizeAndDequantizeV4Grad function. A proof of concept exploit involves using input tensors with incorrect ranks, such as creating tensors with shape [1,1] instead of the required rank 1 (GitHub Advisory).

Mitigation and workarounds

The vulnerability has been patched in TensorFlow 2.5.0 and backported to TensorFlow 2.4.2. Users should upgrade to these patched versions. The fix involves adding proper validation of input tensor ranks before processing (GitHub Commit).

Additional resources

Linux Distribution fix status

Fix availability across major Linux distributions and their releases.

Arch Linux

Fixed

rolling

tensorflow: 2.5.0-1

Fixed

SourceThis report was generated using AI

Related Python vulnerabilities:

CVE ID

Severity

Score

Technologies

Component name

CISA KEV exploit

Has fix

Published date

CVE-2026-61599HIGH8.8
  • Python logoPython
  • djust
NoYesSep 16, 2026
CVE-2026-61596HIGH7.1
  • Python logoPython
  • djust
NoYesSep 16, 2026
CVE-2026-61588MEDIUM6.5
  • Python logoPython
  • djust
NoYesSep 16, 2026
CVE-2026-61589MEDIUM6.3
  • Python logoPython
  • djust
NoYesSep 16, 2026
CVE-2026-61597MEDIUM5.1
  • Python logoPython
  • djust
NoYesSep 16, 2026

Free Vulnerability Assessment

Benchmark your Cloud Security Posture

Evaluate your cloud security practices across 9 security domains to benchmark your risk level and identify gaps in your defenses.

Request assessment

Get a personalized demo

Ready to see Wiz in action?

"Best User Experience I have ever seen, provides full visibility to cloud workloads."
David EstlickCISO
"Wiz provides a single pane of glass to see what is going on in our cloud environments."
Adam FletcherChief Security Officer
"We know that if Wiz identifies something as critical, it actually is."
Greg PoniatowskiHead of Threat and Vulnerability Management