def call(self, hidden_states, **kwargs):
    x = hidden_states[:, 0, :]
    x = self.dropout(x)
    x = self.dense(x)
    x = get_tf_activation(self.config.hidden_act)(x)
    x = self.dropout(x)
    x = self.out_proj(x)
    return x
def call(self, hidden_states, **kwargs):
    x = hidden_states[:, 0, :]
    x = self.dropout(x)
    x = self.dense(x)
    x = get_tf_activation(self.config.hidden_act)(x)
    x = self.dropout(x)
    x = self.out_proj(x)
    return x