code: def _get_pooler(self, units, prefix):
          
          with self.name_scope():
              pooler = nn.Dense(units=units, flatten=False, activation='tanh',
                                prefix=prefix)
          return pooler
docs: Construct pooler.
  
          The pooler slices and projects the hidden output of first token
          in the sequence for segment level classification.