def __init__(self, config, **kwargs):
    super().__init__(**kwargs)
    self.dense = tf.keras.layers.Dense(config.hidden_size, kernel_initializer=get_initializer(config.initializer_range), name='dense')
    classifier_dropout = (config.classifier_dropout if (config.classifier_dropout is not None) else config.hidden_dropout_prob)
    self.dropout = tf.keras.layers.Dropout(classifier_dropout)
    self.out_proj = tf.keras.layers.Dense(config.num_labels, kernel_initializer=get_initializer(config.initializer_range), name='out_proj')
    self.config = config
def __init__(self, config, **kwargs):
    super().__init__(**kwargs)
    self.dense = tf.keras.layers.Dense(config.hidden_size, kernel_initializer=get_initializer(config.initializer_range), name='dense')
    classifier_dropout = (config.classifier_dropout if (config.classifier_dropout is not None) else config.hidden_dropout_prob)
    self.dropout = tf.keras.layers.Dropout(classifier_dropout)
    self.out_proj = tf.keras.layers.Dense(config.num_labels, kernel_initializer=get_initializer(config.initializer_range), name='out_proj')
    self.config = config