experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixing nondeterministic nullptr access.

Browse Source 
PiperOrigin-RevId: 315050926
Change-Id: I6817c077c00118bf336848f0ff578125a4880023
This commit is contained in:
A. Unique TensorFlower 2020-06-05 22:41:51 -07:00 committed by TensorFlower Gardener
parent 2f402cbf23
commit 48ccc08a8b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/sparse_add_op.cc
View File

@ -159,7 +159,9 @@ class SparseAddOp : public OpKernel {
out_indices_mat.chip<0>(i) =
from_a ? a_indices_mat.chip<0>(idx) : b_indices_mat.chip<0>(idx);
}
std::copy_n(out_values.begin(), sum_nnz, &out_values_flat(0));
if (sum_nnz > 0) {
std::copy_n(out_values.begin(), sum_nnz, &out_values_flat(0));
}
ctx->set_output(2, *a_shape);
}
};