libreasr.lib.layers.custom_rnn.get_rnn_impl

libreasr.lib.layers.custom_rnn.get_rnn_impl(device, rnn_type, layer_norm=False)