def setup_method(self):
    super().setup_method()
    self.batch_size = 2
    self.hidden_dim = 2
    self.step_hidden_var = tf.compat.v1.placeholder(shape=(self.batch_size, self.hidden_dim), name='initial_hidden', dtype=tf.float32)
    self.gru_cell = tf.keras.layers.GRUCell(units=self.hidden_dim, activation=tf.nn.tanh, kernel_initializer=tf.constant_initializer(1), recurrent_activation=tf.nn.sigmoid, recurrent_initializer=tf.constant_initializer(1), name='lstm_layer')
def setup_method(self):
    super().setup_method()
    self.batch_size = 2
    self.hidden_dim = 2
    self.step_hidden_var = tf.compat.v1.placeholder(shape=(self.batch_size, self.hidden_dim), name='initial_hidden', dtype=tf.float32)
    self.gru_cell = tf.keras.layers.GRUCell(units=self.hidden_dim, activation=tf.nn.tanh, kernel_initializer=tf.constant_initializer(1), recurrent_activation=tf.nn.sigmoid, recurrent_initializer=tf.constant_initializer(1), name='lstm_layer')