Function func5

src/transforms/bert_ops.py:570–570

()

        original_labels2_token_ids, output_token_len)

    def func5(): return tf.concat([padded_original_labels2_token_ids[:, :-1], tf.fill([count3, 1], eos_token)], axis=1)
    def func6(): return padded_original_labels2_token_ids
    padded_original_labels2_token_ids = tf.cond(tf.equal(original_labels2_pad_mask[0, -1], tf.constant(1)), func5, func6)