aihwkit.nn.modules.rnn package

Analog RNN related modules.

Submodules

• aihwkit.nn.modules.rnn.cells module
  • AnalogGRUCell
    • AnalogGRUCell.forward()
    • AnalogGRUCell.get_zero_state()
  • AnalogLSTMCell
    • AnalogLSTMCell.forward()
    • AnalogLSTMCell.get_zero_state()
  • AnalogLSTMCellCombinedWeight
    • AnalogLSTMCellCombinedWeight.forward()
    • AnalogLSTMCellCombinedWeight.get_zero_state()
  • AnalogVanillaRNNCell
    • AnalogVanillaRNNCell.forward()
    • AnalogVanillaRNNCell.get_zero_state()
  • LSTMState
    • LSTMState.cx
    • LSTMState.hx
• aihwkit.nn.modules.rnn.layers module
  • AnalogBidirRNNLayer
    • AnalogBidirRNNLayer.forward()
    • AnalogBidirRNNLayer.get_zero_state()
  • AnalogRNNLayer
    • AnalogRNNLayer.forward()
    • AnalogRNNLayer.get_zero_state()
  • AnalogReverseRNNLayer
    • AnalogReverseRNNLayer.forward()
    • AnalogReverseRNNLayer.get_zero_state()
    • AnalogReverseRNNLayer.reverse()
• aihwkit.nn.modules.rnn.rnn module
  • AnalogRNN
    • AnalogRNN.forward()
    • AnalogRNN.get_zero_state()
    • AnalogRNN.init_layers()
    • AnalogRNN.reset_parameters()
  • ModularRNN
    • ModularRNN.forward()
    • ModularRNN.get_zero_state()
    • ModularRNN.init_stacked_analog_lstm()