b045ce7d71c64336e34589d20993bbbed2399806,official/nlp/xlnet_modeling.py,TransformerXLModel,call,#TransformerXLModel#Any#,774

Before Change

    else:
      non_tgt_mask = None

    word_emb_k, _ = self.embedding_lookup(inp_k)

    if inp_q is not None:
      if target_mapping is not None:

After Change

            tf.equal(seg_id, tf.constant([data_utils.SEG_ID_CLS]))[:, None],
            tf.equal(cat_id, tf.constant([data_utils.SEG_ID_CLS]))[None, :])
        seg_mat = tf.equal(seg_id[:, None], cat_id[None, :])
        seg_mat = tf.logical_or(cls_mat, seg_mat)
      else:
        seg_mat = tf.logical_not(tf.equal(seg_id[:, None], cat_id[None, :]))
    else:
      seg_mat = None

Instances