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

Prevent dereferencing nullptr in CopyIntoSparseTensor called with empty tensor.

Browse Source 
PiperOrigin-RevId: 323649311
Change-Id: I5f55bbd6a3c6be35a0ba9620e395e447e247e15e
This commit is contained in:
A. Unique TensorFlower 2020-07-28 14:06:14 -07:00 committed by TensorFlower Gardener
parent 33bee8e716
commit 916e0023b9
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/core/util/example_proto_helper.cc
View File

@ -155,11 +155,13 @@ int64 CopyIntoSparseTensor(const Tensor& in, const int batch,
CHECK_EQ(dtype, values->dtype());
// Update indices.
auto ix_t = indices->matrix<int64>();
int64* ix_p = &ix_t(offset, 0);
for (int64 i = 0; i < num_elements; ++i, ix_p += 2) {
*ix_p = batch; // Column 0 stores the batch entry
*(ix_p + 1) = i; // Column 1 stores the index in the batch
if (num_elements > 0) {
auto ix_t = indices->matrix<int64>();
int64* ix_p = &ix_t(offset, 0);
for (int64 i = 0; i < num_elements; ++i, ix_p += 2) {
*ix_p = batch; // Column 0 stores the batch entry
*(ix_p + 1) = i; // Column 1 stores the index in the batch
}
}
// Copy values over.