| 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 |
If a graph node is invalid, TensorFlow can leak memory in the implementation of ImmutableExecutorState::Initialize:
Status s = params_.create_kernel(n->properties(), &item->kernel);
if (!s.ok()) {
item->kernel = nullptr;
s = AttachDef(s, *n);
return s;
}
Here, we set item->kernel to nullptr but it is a simple OpKernel* pointer so the memory that was previously allocated to it would leak.
We have patched the issue in GitHub commit c79ccba517dbb1a0ccb9b01ee3bd2a63748b60dd. 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.
Running one of these products? Scan your site to see whether it exposes a vulnerable version.
Scan my site free