aihwkit_lightning.simulator.parameters package

RPU simulator bindings.

Submodules

• aihwkit_lightning.simulator.parameters.enums module
  • WeightClipType
    • WeightClipType.LAYER_GAUSSIAN
    • WeightClipType.LAYER_GAUSSIAN_PER_CHANNEL
    • WeightClipType.LEARNABLE_PER_CHANNEL
    • WeightClipType.NONE
  • WeightModifierType
    • WeightModifierType.ADD_NORMAL
    • WeightModifierType.ADD_NORMAL_PER_CHANNEL
    • WeightModifierType.DISCRETIZE
    • WeightModifierType.DISCRETIZE_ADD_NORMAL
    • WeightModifierType.DISCRETIZE_ADD_NORMAL_PER_CHANNEL
    • WeightModifierType.DISCRETIZE_PER_CHANNEL
    • WeightModifierType.NONE
  • WeightNoiseInjectionType
    • WeightNoiseInjectionType.ADD_NORMAL
    • WeightNoiseInjectionType.ADD_NORMAL_PER_CHANNEL
    • WeightNoiseInjectionType.NONE
  • WeightQuantizationType
    • WeightQuantizationType.DISCRETIZE
    • WeightQuantizationType.DISCRETIZE_PER_CHANNEL
    • WeightQuantizationType.NONE
• aihwkit_lightning.simulator.parameters.helpers module
• aihwkit_lightning.simulator.parameters.inference module
  • WeightClipParameter
    • WeightClipParameter.sigma
    • WeightClipParameter.type
  • WeightModifierParameter
    • WeightModifierParameter.enable_during_test
    • WeightModifierParameter.noise_type
    • WeightModifierParameter.quantization_type
    • WeightModifierParameter.res
    • WeightModifierParameter.std_dev
    • WeightModifierParameter.type
• aihwkit_lightning.simulator.parameters.io module
  • IOParameters
    • IOParameters.inp_res
    • IOParameters.out_bound
    • IOParameters.out_noise
    • IOParameters.out_noise_per_channel
    • IOParameters.out_res
• aihwkit_lightning.simulator.parameters.mapping module
  • MappableRPU
    • MappableRPU.mapping
  • MappingParameter
    • MappingParameter.max_input_size
• aihwkit_lightning.simulator.parameters.pre_post module
  • InputRangeParameter
    • InputRangeParameter.act_range_momentum
    • InputRangeParameter.decay
    • InputRangeParameter.dynamic
    • InputRangeParameter.enable
    • InputRangeParameter.fast_mode
    • InputRangeParameter.init_from_data
    • InputRangeParameter.init_std_alpha
    • InputRangeParameter.init_value
    • InputRangeParameter.input_min_percentage
    • InputRangeParameter.learn_input_range
  • PrePostProcessingParameter
    • PrePostProcessingParameter.input_range
  • PrePostProcessingRPU
    • PrePostProcessingRPU.pre_post