aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/sparse_concat_op.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/sparse_concat_op.cc')
-rw-r--r--tensorflow/core/kernels/sparse_concat_op.cc9
1 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/sparse_concat_op.cc b/tensorflow/core/kernels/sparse_concat_op.cc
index f813794374..3b2a0cb0f3 100644
--- a/tensorflow/core/kernels/sparse_concat_op.cc
+++ b/tensorflow/core/kernels/sparse_concat_op.cc
@@ -124,9 +124,12 @@ class SparseConcatOp : public OpKernel {
std::vector<sparse::SparseTensor> sp_inputs;
for (int i = 0; i < N; ++i) {
const TensorShape current_shape(shapes[i].vec<int64>());
- sp_inputs.emplace_back(tensor::DeepCopy(inds[i]),
- tensor::DeepCopy(vals[i]), current_shape,
- std_order);
+ sparse::SparseTensor tensor;
+ OP_REQUIRES_OK(context,
+ sparse::SparseTensor::Create(
+ tensor::DeepCopy(inds[i]), tensor::DeepCopy(vals[i]),
+ current_shape, std_order, &tensor));
+ sp_inputs.push_back(std::move(tensor));
sp_inputs[i].Reorder<T>(concat_order);
}