Skip to content

GitLab

Commit a977b4e1 authored by Philip Pham's avatar Philip Pham Committed by A. Unique TensorFlower
Browse files

Fix typo in TransformerEncoderBlock docstring 

PiperOrigin-RevId: 383660517
parent f3022ed8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/keras_nlp/layers/transformer_encoder_block.py
View file @ a977b4e1
...@@ -249,7 +249,7 @@ class TransformerEncoderBlock(tf.keras.layers.Layer): ...@@ -249,7 +249,7 @@ class TransformerEncoderBlock(tf.keras.layers.Layer):
attention. attention.
Returns: Returns:
An ouput tensor with the same dimensions as input/query tensor. An output tensor with the same dimensions as input/query tensor.
""" """
if isinstance(inputs, (list, tuple)): if isinstance(inputs, (list, tuple)):
if len(inputs) == 2: if len(inputs) == 2:
......
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