libreasr.lib.layers.custom_rnnΒΆ

Functions

copy_weights

get_initial_state

get_rnn_impl

get_weight_attrs

Classes

CustomCPURNN

Initializes internal Module state, shared by both nn.Module and ScriptModule.

CustomGPURNN

Initializes internal Module state, shared by both nn.Module and ScriptModule.

CustomRNN

Initializes internal Module state, shared by both nn.Module and ScriptModule.