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

Fix typo in TransformerEncoderBlock docstring

PiperOrigin-RevId: 383660517
parent e71d67c3
......@@ -249,7 +249,7 @@ class TransformerEncoderBlock(tf.keras.layers.Layer):
attention.
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 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