TensorFlow is an end-to-end open source platform for machine learning. An attacker can trigger a denial of service via a CHECK-fail in tf.raw_ops.QuantizeAndDequantizeV4Grad. This is because the implementation does not validate the rank of the input_* tensors. In turn, this results in the tensors being passes as they are to QuantizeAndDequantizePerChannelGradientImpl. However, the vec<T> method, requires the rank to 1 and triggers a CHECK failure otherwise. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2 as this is the only other affected version.
[
{
"source": "https://github.com/tensorflow/tensorflow/commit/20431e9044cf2ad3c0323c34888b192f3289af6b",
"id": "CVE-2021-29544-a2f93e61",
"digest": {
"line_hashes": [
"176068773716387915431679173772228483723",
"195008316136258450943681231606746130387",
"95720392910609131018701593023343122866",
"3060895597940912716767874583125497482",
"182232495582687097407204052409773180322"
],
"threshold": 0.9
},
"signature_version": "v1",
"deprecated": false,
"signature_type": "Line",
"target": {
"file": "tensorflow/core/kernels/quantize_and_dequantize_op.cc"
}
}
]