aihwkit.nn.low_precision_modules package

Submodules

• aihwkit.nn.low_precision_modules.conversion_utils module
  • append_default_conversions()
  • get_module_args()
• aihwkit.nn.low_precision_modules.quantization_states module
  • enable_quant_states()
  • estimate_act_ranges()
  • estimate_ranges()
  • estimate_ranges_train()
  • fix_act_ranges()
  • fix_ranges()
  • fix_weight_ranges()
  • full_precision()
  • full_precision_acts()
  • full_precision_weights()
  • learn_ranges()
  • quantized()
  • quantized_acts()
  • quantized_weights()
  • reset_act_ranges()
  • set_quant_state()
• aihwkit.nn.low_precision_modules.quantized_base_modules module
  • QuantBatchNorm2d
    • QuantBatchNorm2d.forward()
  • QuantConv2d
    • QuantConv2d.run_forward()
  • QuantEmbedding
    • QuantEmbedding.run_forward()
  • QuantLayerNorm
    • QuantLayerNorm.run_forward()
  • QuantLinear
    • QuantLinear.run_forward()
• aihwkit.nn.low_precision_modules.quantized_input_module module
  • QuantizedInputModule
    • QuantizedInputModule.forward()