Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "8c9670375f80b3a5bdc688912809f7f65bfc3487"
Commit 2cad427b authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

internal change 

PiperOrigin-RevId: 334829363
parent ad0b28c6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/nlp/keras_nlp/layers/masked_lm.py
View file @ 2cad427b
......@@ -81,8 +81,7 @@ class MaskedLM(tf.keras.layers.Layer):
lm_data = self.layer_norm(lm_data)
lm_data = tf.matmul(lm_data, self.embedding_table, transpose_b=True)
logits = tf.nn.bias_add(lm_data, self.bias)
masked_positions_length = masked_positions.shape[1] or tf.shape(
masked_positions_length = masked_positions.shape.as_list()[1] or tf.shape(
masked_positions)[1]
logits = tf.reshape(logits,
[-1, masked_positions_length, self._vocab_size])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment