def test_rnn_block():
utils.block_basic_exam(
basic.RNNBlock(),
tf.keras.Input(shape=(32, 10), dtype=tf.float32),
[
'bidirectional',
'layer_type',
'num_layers',
])
def build(self, hp, inputs=None):
input_node = nest.flatten(inputs)[0]
output_node = input_node
output_node = basic.RNNBlock().build(hp, output_node)
return output_node
