Function func2

src/transforms/bert_ops.py:504–504

()

  
    def func1(): return tf.concat([padded_token_ids[:, :-1], tf.fill([count, 1], eos_token)], axis=1)
    def func2(): return padded_token_ids
    padded_token_ids = tf.cond(tf.equal(pad_mask[0, -1], tf.constant(1)), func1, func2)
  
    padded_token_ids = tf.concat([padded_token_ids, tf.fill([count, 1], cls_token)], axis=1)