aihwkit.nn.modules package

Neural network modules.

Subpackages

• aihwkit.nn.modules.rnn package
  • Submodules
    • aihwkit.nn.modules.rnn.cells module
      • AnalogGRUCell
      • AnalogLSTMCell
      • AnalogLSTMCellCombinedWeight
      • AnalogVanillaRNNCell
      • LSTMState
    • aihwkit.nn.modules.rnn.layers module
      • AnalogBidirRNNLayer
      • AnalogRNNLayer
      • AnalogReverseRNNLayer
    • aihwkit.nn.modules.rnn.rnn module
      • AnalogRNN
      • ModularRNN

Submodules

• aihwkit.nn.modules.base module
  • AnalogLayerBase
    • AnalogLayerBase.IS_CONTAINER
    • AnalogLayerBase.analog_layers()
    • AnalogLayerBase.analog_modules()
    • AnalogLayerBase.analog_tile_count()
    • AnalogLayerBase.analog_tiles()
    • AnalogLayerBase.apply_to_analog_layers()
    • AnalogLayerBase.apply_to_analog_tiles()
    • AnalogLayerBase.drift_analog_weights()
    • AnalogLayerBase.extra_repr()
    • AnalogLayerBase.get_analog_tile_devices()
    • AnalogLayerBase.get_weights()
    • AnalogLayerBase.load_state_dict()
    • AnalogLayerBase.named_analog_layers()
    • AnalogLayerBase.named_analog_modules()
    • AnalogLayerBase.named_analog_tiles()
    • AnalogLayerBase.prepare_for_ddp()
    • AnalogLayerBase.program_analog_weights()
    • AnalogLayerBase.remap_analog_weights()
    • AnalogLayerBase.replace_rpu_config()
    • AnalogLayerBase.set_weights()
    • AnalogLayerBase.unregister_parameter()
• aihwkit.nn.modules.container module
  • AnalogContainerBase
    • AnalogContainerBase.IS_CONTAINER
    • AnalogContainerBase.get_weights()
    • AnalogContainerBase.set_weights()
  • AnalogSequential
    • AnalogSequential.from_digital()
    • AnalogSequential.to_digital()
  • AnalogWrapper
    • AnalogWrapper.SUBCLASSES
    • AnalogWrapper.from_digital()
    • AnalogWrapper.to_digital()
• aihwkit.nn.modules.conv module
  • AnalogConv1d
    • AnalogConv1d.NEEDS_INDEXED
    • AnalogConv1d.from_digital()
    • AnalogConv1d.get_tile_size()
    • AnalogConv1d.to_digital()
  • AnalogConv2d
    • AnalogConv2d.from_digital()
    • AnalogConv2d.get_tile_size()
    • AnalogConv2d.to_digital()
  • AnalogConv3d
    • AnalogConv3d.NEEDS_INDEXED
    • AnalogConv3d.from_digital()
    • AnalogConv3d.get_tile_size()
    • AnalogConv3d.to_digital()
• aihwkit.nn.modules.conv_mapped module
  • AnalogConv1dMapped
    • AnalogConv1dMapped.from_digital()
    • AnalogConv1dMapped.get_tile_size()
    • AnalogConv1dMapped.to_digital()
  • AnalogConv2dMapped
    • AnalogConv2dMapped.from_digital()
    • AnalogConv2dMapped.get_tile_size()
    • AnalogConv2dMapped.to_digital()
  • AnalogConv3dMapped
    • AnalogConv3dMapped.from_digital()
    • AnalogConv3dMapped.get_tile_size()
    • AnalogConv3dMapped.to_digital()
• aihwkit.nn.modules.linear module
  • AnalogLinear
    • AnalogLinear.forward()
    • AnalogLinear.from_digital()
    • AnalogLinear.reset_parameters()
    • AnalogLinear.to_digital()
• aihwkit.nn.modules.linear_mapped module
  • AnalogLinearMapped
    • AnalogLinearMapped.from_digital()