GitLab

num_output_classes -> pooled_output_dim
classification_layer_initializer -> pooler_layer_initializer

PiperOrigin-RevId: 305550415