diff options
Diffstat (limited to 'tensorflow/core/kernels/ctc_loss_op.cc')
-rw-r--r-- | tensorflow/core/kernels/ctc_loss_op.cc | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/ctc_loss_op.cc b/tensorflow/core/kernels/ctc_loss_op.cc index 05d0169b11..426382edec 100644 --- a/tensorflow/core/kernels/ctc_loss_op.cc +++ b/tensorflow/core/kernels/ctc_loss_op.cc @@ -42,6 +42,8 @@ class CTCLossOp : public OpKernel { &preprocess_collapse_repeated_)); OP_REQUIRES_OK(ctx, ctx->GetAttr("ctc_merge_repeated", &ctc_merge_repeated_)); + OP_REQUIRES_OK(ctx, ctx->GetAttr("ignore_longer_outputs_than_inputs", + &ignore_longer_outputs_than_inputs_)); } void Compute(OpKernelContext* ctx) override { @@ -150,12 +152,15 @@ class CTCLossOp : public OpKernel { OP_REQUIRES_OK(ctx, ctc_loss_calculator.CalculateLoss( seq_len_t, labels_t, input_list_t, preprocess_collapse_repeated_, ctc_merge_repeated_, - &loss_t, &gradient_list_t, &workers)); + ignore_longer_outputs_than_inputs_, &loss_t, + &gradient_list_t, &workers)); } private: bool preprocess_collapse_repeated_; bool ctc_merge_repeated_; + bool ignore_longer_outputs_than_inputs_; + TF_DISALLOW_COPY_AND_ASSIGN(CTCLossOp); }; |