Internal change

PiperOrigin-RevId: 408650343

official/nlp/modeling/networks/bert_dense_encoder.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Transformer-based BERT encoder network."""
+"""Transformer-based BERT encoder network with dense features as inputs."""
 # pylint: disable=g-classes-have-attributes
 
 from typing import Any, Callable, Optional, Union

official/nlp/modeling/networks/bert_dense_encoder_test.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Tests for transformer-based bert encoder network."""
+"""Tests for transformer-based bert encoder network with dense features as inputs."""
 
 # Import libraries
 from absl.testing import parameterized