diff options
author | 2017-07-31 01:07:41 -0700 | |
---|---|---|
committer | 2017-07-31 01:10:50 -0700 | |
commit | f19bb3bebfb78c0adc7346e7951d39b5bf219a12 (patch) | |
tree | 12b83ddb3e06d1693281708d3340cb13141351a4 /tensorflow/core/kernels/ctc_loss_op.cc | |
parent | e17650b69853d2f5706a51ac775375100408abb3 (diff) |
Prevent ctc_loss op from segfaulting when given empty batch.
PiperOrigin-RevId: 163663460
Diffstat (limited to 'tensorflow/core/kernels/ctc_loss_op.cc')
-rw-r--r-- | tensorflow/core/kernels/ctc_loss_op.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/ctc_loss_op.cc b/tensorflow/core/kernels/ctc_loss_op.cc index 426382edec..a1f6001914 100644 --- a/tensorflow/core/kernels/ctc_loss_op.cc +++ b/tensorflow/core/kernels/ctc_loss_op.cc @@ -88,6 +88,9 @@ class CTCLossOp : public OpKernel { labels_indices->shape().DebugString(), " vs. ", labels_values->shape().DebugString())); + OP_REQUIRES(ctx, batch_size != 0, + errors::InvalidArgument("batch_size must not be 0")); + TensorShape labels_shape({batch_size, max_time}); std::vector<int64> order{0, 1}; sparse::SparseTensor labels_sp(*labels_indices, *labels_values, |