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