QID 981509
QID 981509: Python (pip) Security Update for tensorflow-gpu (GHSA-95xm-g58g-3p88)
Security update has been released for tensorflow-gpu,tensorflow,tensorflow-cpu to fix the vulnerability.
Note: The preceding description block is extracted directly from the security advisory. Using automation, we have attempted to clean and format it as much as possible without introducing additional issues.
The implementation of `tf.raw_ops.SparseReshape` can be made to trigger an integral division by 0 exception:
```python
import tensorflow as tf
tf.raw_ops.SparseReshape(
input_indices = np.ones((1,3)),
input_shape = np.array([1,1,0]),
new_shape = np.array([1,0]))
```
The [implementation](https://github.com/tensorflow/tensorflow/blob/8d72537c6abf5a44103b57b9c2e22c14f5f49698/tensorflow/core/kernels/reshape_util.cc#L176-L181) calls the reshaping functor whenever there is at least an index in the input but does not check that shape of the input or the target shape have both a non-zero number of elements.
The [reshape functor](https://github.com/tensorflow/tensorflow/blob/8d72537c6abf5a44103b57b9c2e22c14f5f49698/tensorflow/core/kernels/reshape_util.cc#L40-L78) blindly divides by the dimensions of the target shape. Hence, if this is not checked, code will result in a division by 0.
The fix will be included in TensorFlow 2.6.0. We will also cherrypick this commit on TensorFlow 2.5.1 as this is the other affected version.
- GHSA-95xm-g58g-3p88 -
github.com/advisories/GHSA-95xm-g58g-3p88
CVEs related to QID 981509
| Advisory ID | Software | Component | Link |
|---|---|---|---|
| GHSA-95xm-g58g-3p88 | tensorflow |
|
|
| GHSA-95xm-g58g-3p88 | tensorflow-cpu |
|
|
| GHSA-95xm-g58g-3p88 | tensorflow-gpu |
|