Update typo in `compute_output_shape`
Since the method name is `compute_output_shape` returns should be `an output shape tuple` not `an input shape tuple`.
@@ -841,7 +841,7 @@ class Layer(tf.Module, version_utils.LayerVersionSelector):
instead of an integer.
Returns:
- An input shape tuple.
+ An output shape tuple.
"""
if tf.executing_eagerly():
# In this case we build the model first in order to do shape
