`CHECK`-fails due to attempting to build a reference tensor

Overview

Impact

A malicious user can cause a denial of service by altering a SavedModel such that Grappler optimizer would attempt to build a tensor using a reference dtype. This would result in a crash due to a CHECK-fail in the Tensor constructor as reference types are not allowed.

Patches

We have patched the issue in GitHub commit 6b5adc0877de832b2a7c189532dbbbc64622eeb6. 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.

For more information

Please consult our security guide for more information regarding the security model and how to contact us with issues and questions.

Affected packages

• tensorflow < 2.5.3
• tensorflow >= 2.6.0, < 2.6.3
• tensorflow >= 2.7.0, < 2.7.1
• tensorflow-cpu < 2.5.3
• tensorflow-cpu >= 2.6.0, < 2.6.3
• tensorflow-cpu >= 2.7.0, < 2.7.1
• tensorflow-gpu < 2.5.3
• tensorflow-gpu >= 2.6.0, < 2.6.3
• tensorflow-gpu >= 2.7.0, < 2.7.1

Remediation

Upgrade to a patched release:

• tensorflow 2.5.3
• tensorflow 2.6.3
• tensorflow 2.7.1
• tensorflow-cpu 2.5.3
• tensorflow-cpu 2.6.3
• tensorflow-cpu 2.7.1
• tensorflow-gpu 2.5.3
• tensorflow-gpu 2.6.3
• tensorflow-gpu 2.7.1