Skip to content
OpenAdvisories
Low · 2.5 CVE-2021-29596 BIT-tensorflow-2021-29596EUVD-2021-0337GHSA-4vrf-ff7v-hpgrPYSEC-2021-233PYSEC-2021-524PYSEC-2021-722

Division by zero in TFLite's implementation of `EmbeddingLookup`

Published May 14, 2021 · Updated Aug 3, 2024 · Assigned by GitHub_M

Vendor: TensorFlow Product: TensorFlow Package: tensorflow Package: tensorflow-cpu Package: tensorflow-gpu Ecosystem: PyPI

Description

TensorFlow is an end-to-end open source platform for machine learning. The implementation of the `EmbeddingLookup` TFLite operator is vulnerable to a division by zero error(https://github.com/tensorflow/tensorflow/blob/e4b29809543b250bc9b19678ec4776299dd569ba/tensorflow/lite/kernels/embedding_lookup.cc#L73-L74). An attacker can craft a model such that the first dimension of the `value` input is 0. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2, TensorFlow 2.3.3, TensorFlow 2.2.3 and TensorFlow 2.1.4, as these are also affected and still in supported range.

Affected products

VendorProductAffectedFixed
tensorflow tensorflow < 2.1.4, >= 2.2.0, < 2.2.3, >= 2.3.0, < 2.3.3, >= 2.4.0, < 2.4.2 —

Affected packages

EcosystemPackageAffectedFixedRecord
PyPI tensorflow < 2.1.4 2.1.4 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow >= 2.2.0, < 2.2.3 2.2.3 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow >= 2.3.0, < 2.3.3 2.3.3 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow >= 2.4.0, < 2.4.2 2.4.2 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow < 2.1.4 · >= 2.2.0, < 2.2.3 · >= 2.3.0, < 2.3.3 · >= 2.4.0, < 2.4.2 2.1.4, 2.2.3, 2.3.3, 2.4.2 PYSEC-2021-233
PyPI tensorflow-cpu < 2.1.4 2.1.4 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow-cpu >= 2.2.0, < 2.2.3 2.2.3 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow-cpu >= 2.3.0, < 2.3.3 2.3.3 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow-cpu >= 2.4.0, < 2.4.2 2.4.2 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow-cpu < 2.1.4 · >= 2.2.0, < 2.2.3 · >= 2.3.0, < 2.3.3 · >= 2.4.0, < 2.4.2 2.1.4, 2.2.3, 2.3.3, 2.4.2 PYSEC-2021-524
PyPI tensorflow-gpu < 2.1.4 2.1.4 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow-gpu >= 2.2.0, < 2.2.3 2.2.3 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow-gpu >= 2.3.0, < 2.3.3 2.3.3 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow-gpu >= 2.4.0, < 2.4.2 2.4.2 GHSA-4vrf-ff7v-hpgr
PyPI tensorflow-gpu < 2.1.4 · >= 2.2.0, < 2.2.3 · >= 2.3.0, < 2.3.3 · >= 2.4.0, < 2.4.2 2.1.4, 2.2.3, 2.3.3, 2.4.2 PYSEC-2021-722

References