diff options
Diffstat (limited to 'tensorflow/core/kernels/sparse_reduce_op.cc')
-rw-r--r-- | tensorflow/core/kernels/sparse_reduce_op.cc | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/sparse_reduce_op.cc b/tensorflow/core/kernels/sparse_reduce_op.cc index 9e60791f97..a465564739 100644 --- a/tensorflow/core/kernels/sparse_reduce_op.cc +++ b/tensorflow/core/kernels/sparse_reduce_op.cc @@ -172,8 +172,10 @@ class SparseReduceOp : public OpKernel { // making deep copies here. Remove this if/when we change Reorder()'s // semantics. const auto shape_vec = shape_t->vec<int64>(); - SparseTensor sp(tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), - TensorShape(shape_vec)); + SparseTensor sp; + OP_REQUIRES_OK(ctx, SparseTensor::Create( + tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), + TensorShape(shape_vec), &sp)); ReduceDetails reduction = SparseTensorReduceHelper( sp, reduction_axes_t->flat<int32>(), keep_dims_); @@ -260,8 +262,10 @@ class SparseReduceSparseOp : public OpKernel { OP_REQUIRES_OK(ctx, ValidateInputs(shape_t, reduction_axes_t)); - SparseTensor sp(tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), - TensorShape(shape_t->vec<int64>())); + SparseTensor sp; + OP_REQUIRES_OK(ctx, SparseTensor::Create(tensor::DeepCopy(*indices_t), + tensor::DeepCopy(*values_t), + TensorShape(shape_t->vec<int64>()), &sp)); ReduceDetails reduction = SparseTensorReduceHelper( sp, reduction_axes_t->flat<int32>(), keep_dims_); |