diff options
Diffstat (limited to 'tensorflow/core/kernels/ctc_loss_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/ctc_loss_op.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/ctc_loss_op.cc b/tensorflow/core/kernels/ctc_loss_op.cc index b38d838bf1..fb375ee4b3 100644 --- a/tensorflow/core/kernels/ctc_loss_op.cc +++ b/tensorflow/core/kernels/ctc_loss_op.cc @@ -100,8 +100,10 @@ class CTCLossOp : public OpKernel { TensorShape labels_shape({batch_size, max_label_len}); std::vector<int64> order{0, 1}; - sparse::SparseTensor labels_sp(*labels_indices, *labels_values, - labels_shape, order); + sparse::SparseTensor labels_sp; + OP_REQUIRES_OK( + ctx, sparse::SparseTensor::Create(*labels_indices, *labels_values, + labels_shape, order, &labels_sp)); Status labels_sp_valid = labels_sp.IndicesValid(); OP_REQUIRES(ctx, labels_sp_valid.ok(), |