aihwkit.simulator.digital_low_precision package

Submodules

• aihwkit.simulator.digital_low_precision.base_quantized_classes module
  • FP32Acts
    • FP32Acts.forward()
    • FP32Acts.reset_ranges()
  • QuantizedActivation
    • QuantizedActivation.forward()
    • QuantizedActivation.quantize_activations()
  • QuantizedModule
    • QuantizedModule.caching
    • QuantizedModule.estimate_ranges()
    • QuantizedModule.estimate_ranges_train()
    • QuantizedModule.extra_repr()
    • QuantizedModule.fix_ranges()
    • QuantizedModule.full_precision()
    • QuantizedModule.full_precision_acts()
    • QuantizedModule.full_precision_weights()
    • QuantizedModule.learn_ranges()
    • QuantizedModule.quantized()
    • QuantizedModule.quantized_acts()
    • QuantizedModule.quantized_weights()
    • QuantizedModule.train()
• aihwkit.simulator.digital_low_precision.base_quantized_model module
  • QuantizedModel
    • QuantizedModel.estimate_act_ranges()
    • QuantizedModel.estimate_ranges()
    • QuantizedModel.estimate_ranges_train()
    • QuantizedModel.fix_act_ranges()
    • QuantizedModel.fix_ranges()
    • QuantizedModel.fix_weight_ranges()
    • QuantizedModel.full_precision()
    • QuantizedModel.full_precision_acts()
    • QuantizedModel.full_precision_weights()
    • QuantizedModel.learn_ranges()
    • QuantizedModel.quantized()
    • QuantizedModel.quantized_acts()
    • QuantizedModel.quantized_weights()
    • QuantizedModel.reset_act_ranges()
    • QuantizedModel.set_quant_state()
• aihwkit.simulator.digital_low_precision.config_utils module
  • CrossEntropyEstimatorParams
  • CurrentMinMaxEstimatorParams
    • CurrentMinMaxEstimatorParams.percentile
  • MSEEstimatorParams
    • MSEEstimatorParams.num_candidates
    • MSEEstimatorParams.range_margin
    • MSEEstimatorParams.range_opt_method
  • RunningMinMaxEstimatorParams
    • RunningMinMaxEstimatorParams.momentum
  • convert_act_config_to_kwargs_dict()
  • convert_configs_to_kwargs_dict()
  • convert_weight_config_to_kwargs_dict()
• aihwkit.simulator.digital_low_precision.hijacker module
  • QuantizationHijacker
    • QuantizationHijacker.forward()
    • QuantizationHijacker.get_params()
    • QuantizationHijacker.get_weight_bias()
    • QuantizationHijacker.quantize_activations()
    • QuantizationHijacker.run_forward()
• aihwkit.simulator.digital_low_precision.quantization_manager module
  • Qstates
    • Qstates.estimate_ranges
    • Qstates.estimate_ranges_train
    • Qstates.fix_ranges
    • Qstates.learn_ranges
  • QuantizationManager
    • QuantizationManager.estimate_ranges()
    • QuantizationManager.estimate_ranges_train()
    • QuantizationManager.extra_repr()
    • QuantizationManager.fix_ranges()
    • QuantizationManager.forward()
    • QuantizationManager.is_learning()
    • QuantizationManager.learn_ranges()
    • QuantizationManager.n_bits
    • QuantizationManager.quantizer
    • QuantizationManager.reset_ranges()
    • QuantizationManager.set_quant_range()
• aihwkit.simulator.digital_low_precision.quantizers module
  • AsymmetricUniformQuantizer
    • AsymmetricUniformQuantizer.delta
    • AsymmetricUniformQuantizer.forward()
    • AsymmetricUniformQuantizer.int_max
    • AsymmetricUniformQuantizer.int_min
    • AsymmetricUniformQuantizer.is_initialized
    • AsymmetricUniformQuantizer.make_range_trainable()
    • AsymmetricUniformQuantizer.reset()
    • AsymmetricUniformQuantizer.scale
    • AsymmetricUniformQuantizer.set_quant_range()
    • AsymmetricUniformQuantizer.symmetric
    • AsymmetricUniformQuantizer.to_integer_forward()
    • AsymmetricUniformQuantizer.x_max
    • AsymmetricUniformQuantizer.x_min
    • AsymmetricUniformQuantizer.zero_float
    • AsymmetricUniformQuantizer.zero_point
  • FloorStraightThrough
    • FloorStraightThrough.backward()
    • FloorStraightThrough.forward()
  • QMethodMap
    • QMethodMap.cls
    • QMethodMap.value
  • QMethods
    • QMethods.asymmetric_uniform
    • QMethods.cls
    • QMethods.list()
    • QMethods.symmetric_uniform
  • QuantizerBase
    • QuantizerBase.extra_repr()
    • QuantizerBase.forward()
    • QuantizerBase.is_initialized
    • QuantizerBase.reset()
    • QuantizerBase.set_quant_range()
    • QuantizerBase.symmetric
    • QuantizerBase.x_max
    • QuantizerBase.x_min
  • QuantizerNotInitializedError
  • RoundStraightThrough
    • RoundStraightThrough.backward()
    • RoundStraightThrough.forward()
  • SymmetricUniformQuantizer
    • SymmetricUniformQuantizer.delta
    • SymmetricUniformQuantizer.forward()
    • SymmetricUniformQuantizer.int_max
    • SymmetricUniformQuantizer.int_min
    • SymmetricUniformQuantizer.is_initialized
    • SymmetricUniformQuantizer.make_range_trainable()
    • SymmetricUniformQuantizer.reset()
    • SymmetricUniformQuantizer.scale
    • SymmetricUniformQuantizer.set_quant_range()
    • SymmetricUniformQuantizer.signed
    • SymmetricUniformQuantizer.symmetric
    • SymmetricUniformQuantizer.to_integer_forward()
    • SymmetricUniformQuantizer.x_max
    • SymmetricUniformQuantizer.x_min
    • SymmetricUniformQuantizer.zero_point
• aihwkit.simulator.digital_low_precision.range_estimators module
  • AllMinMaxEstimator
    • AllMinMaxEstimator.forward()
  • CrossEntropyEstimator
    • CrossEntropyEstimator.loss_fx()
  • CurrentMinMaxEstimator
    • CurrentMinMaxEstimator.forward()
  • MSE_Estimator
    • MSE_Estimator.forward()
    • MSE_Estimator.golden_asym_range_loss()
    • MSE_Estimator.golden_asym_shift_loss()
    • MSE_Estimator.golden_sym_loss()
    • MSE_Estimator.loss_fx()
    • MSE_Estimator.optimization_method
    • MSE_Estimator.quantize()
    • MSE_Estimator.reset()
    • MSE_Estimator.step_size
  • NoDataPassedError
  • OptMethod
    • OptMethod.golden_section
    • OptMethod.grid
    • OptMethod.list()
  • RangeEstimatorBase
    • RangeEstimatorBase.forward()
    • RangeEstimatorBase.reset()
  • RangeEstimatorMap
    • RangeEstimatorMap.cls
    • RangeEstimatorMap.value
  • RangeEstimators
    • RangeEstimators.MSE
    • RangeEstimators.allminmax
    • RangeEstimators.cls
    • RangeEstimators.cross_entropy
    • RangeEstimators.current_minmax
    • RangeEstimators.list()
    • RangeEstimators.running_minmax
  • RunningMinMaxEstimator
    • RunningMinMaxEstimator.forward()
• aihwkit.simulator.digital_low_precision.utils module
  • to_numpy()