From 48ccc08a8b7923d41845e3dcd8e191e5bc1a6b51 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 5 Jun 2020 22:41:51 -0700 Subject: [PATCH] Fixing nondeterministic nullptr access. PiperOrigin-RevId: 315050926 Change-Id: I6817c077c00118bf336848f0ff578125a4880023 --- tensorflow/core/kernels/sparse_add_op.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/core/kernels/sparse_add_op.cc b/tensorflow/core/kernels/sparse_add_op.cc index d16317af671..0cf40a709a3 100644 --- a/tensorflow/core/kernels/sparse_add_op.cc +++ b/tensorflow/core/kernels/sparse_add_op.cc @@ -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); } };