diff options
Diffstat (limited to 'tensorflow/core/kernels/sparse_to_dense_op.cc')
-rw-r--r-- | tensorflow/core/kernels/sparse_to_dense_op.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/sparse_to_dense_op.cc b/tensorflow/core/kernels/sparse_to_dense_op.cc index ba3da21a43..f79a4d0494 100644 --- a/tensorflow/core/kernels/sparse_to_dense_op.cc +++ b/tensorflow/core/kernels/sparse_to_dense_op.cc @@ -119,8 +119,10 @@ class SparseToDense : public OpKernel { // Assume SparseTensor is lexicographically sorted. gtl::InlinedVector<int64, 8> order(output->shape().dims()); std::iota(order.begin(), order.end(), 0); - sparse::SparseTensor st(indices_shaped, sparse_values_b, output->shape(), - order); + sparse::SparseTensor st; + OP_REQUIRES_OK(c, + sparse::SparseTensor::Create(indices_shaped, sparse_values_b, + output->shape(), order, &st)); if (validate_indices_) { OP_REQUIRES_OK(c, st.IndicesValid()); |