def __init__(self, n_state, config, **kwargs):
    super(TFMLP, self).__init__(**kwargs)
    nx = config.n_embd
    self.c_fc = TFConv1D(n_state, nx, initializer_range=config.initializer_range, name='c_fc')
    self.c_proj = TFConv1D(nx, n_state, initializer_range=config.initializer_range, name='c_proj')
    self.act = gelu
    self.dropout = tf.keras.layers.Dropout(config.resid_pdrop)
def __init__(self, n_state, config, **kwargs):
    super(TFMLP, self).__init__(**kwargs)
    nx = config.n_embd
    self.c_fc = TFConv1D(n_state, nx, initializer_range=config.initializer_range, name='c_fc')
    self.c_proj = TFConv1D(nx, n_state, initializer_range=config.initializer_range, name='c_proj')
    self.act = gelu
    self.dropout = tf.keras.layers.Dropout(config.resid_pdrop)