| Package | Affected versions | Patched versions |
|---|---|---|
| tensorflow | ≥ 2.6.0, < 2.6.3≥ 2.7.0, < 2.7.1< 2.5.3 | 2.6.32.7.12.5.3 |
| tensorflow-cpu | ≥ 2.6.0, < 2.6.3≥ 2.7.0, < 2.7.1< 2.5.3 | 2.6.32.7.12.5.3 |
| tensorflow-gpu | ≥ 2.6.0, < 2.6.3≥ 2.7.0, < 2.7.1< 2.5.3 | 2.6.32.7.12.5.3 |
| tensorflow | = 2.7.0≥ 2.6.0, ≤ 2.6.2≤ 2.5.2 |
An attacker can craft a TFLite model that would trigger a division by zero in the implementation of depthwise convolutions.
The parameters of the convolution can be user controlled and are also used within a division operation to determine the size of the padding that needs to be added before applying the convolution. There is no check before this division that the divisor is stricly positive.
We have patched the issue in GitHub commit e5b0eec199c2d03de54fd6a7fd9275692218e2bc.
The fix will be included in TensorFlow 2.8.0. We will also cherrypick this commit on TensorFlow 2.7.1, TensorFlow 2.6.3, and TensorFlow 2.5.3, as these are also affected and still in supported range.
Please consult our security guide for more information regarding the security model and how to contact us with issues and questions.
This vulnerability has been reported by Wang Xuan of Qihoo 360 AIVul Team.
Running one of these products? Scan your site to see whether it exposes a vulnerable version.
Scan my site free