aihwkit.simulator.parameters package

RPU simulator bindings.

Submodules

• aihwkit.simulator.parameters.base module
  • RPUConfigBase
    • RPUConfigBase.as_bindings()
    • RPUConfigBase.compatible_with()
    • RPUConfigBase.create_tile()
    • RPUConfigBase.get_default_tile_module_class()
    • RPUConfigBase.runtime
    • RPUConfigBase.tile_class
• aihwkit.simulator.parameters.enums module
  • AnalogMVType
    • AnalogMVType.BIT_WISE
    • AnalogMVType.IDEAL
    • AnalogMVType.ONE_PASS
    • AnalogMVType.POS_NEG_SEPARATE
    • AnalogMVType.POS_NEG_SEPARATE_DIGITAL_SUM
    • AnalogMVType.SPLIT_MODE
  • BoundManagementType
    • BoundManagementType.ITERATIVE
    • BoundManagementType.ITERATIVE_WORST_CASE
    • BoundManagementType.NONE
    • BoundManagementType.SHIFT
  • CountLRFeedbackPolicy
    • CountLRFeedbackPolicy.EXTERN
    • CountLRFeedbackPolicy.NONE
    • CountLRFeedbackPolicy.PAST_MOMENTUM
  • NoiseManagementType
    • NoiseManagementType.ABS_MAX
    • NoiseManagementType.ABS_MAX_NP_SUM
    • NoiseManagementType.AVERAGE_ABS_MAX
    • NoiseManagementType.CONSTANT
    • NoiseManagementType.MAX
    • NoiseManagementType.NONE
  • PulseType
    • PulseType.DETERMINISTIC_IMPLICIT
    • PulseType.MEAN_COUNT
    • PulseType.NONE
    • PulseType.NONE_WITH_DEVICE
    • PulseType.STOCHASTIC
    • PulseType.STOCHASTIC_COMPRESSED
  • RPUDataType
    • RPUDataType.DOUBLE
    • RPUDataType.FLOAT
    • RPUDataType.HALF
    • RPUDataType.as_torch()
  • VectorUnitCellUpdatePolicy
    • VectorUnitCellUpdatePolicy.ALL
    • VectorUnitCellUpdatePolicy.SINGLE_FIXED
    • VectorUnitCellUpdatePolicy.SINGLE_RANDOM
    • VectorUnitCellUpdatePolicy.SINGLE_SEQUENTIAL
  • WeightClipType
    • WeightClipType.AVERAGE_CHANNEL_MAX
    • WeightClipType.FIXED_VALUE
    • WeightClipType.LAYER_GAUSSIAN
    • WeightClipType.NONE
  • WeightModifierType
    • WeightModifierType.ADD_NORMAL
    • WeightModifierType.COPY
    • WeightModifierType.DISCRETIZE
    • WeightModifierType.DISCRETIZE_ADD_NORMAL
    • WeightModifierType.DOREFA
    • WeightModifierType.DROP_CONNECT
    • WeightModifierType.MULT_NORMAL
    • WeightModifierType.NONE
    • WeightModifierType.PCM_NOISE
    • WeightModifierType.POLY
    • WeightModifierType.PROG_NOISE
  • WeightNoiseType
    • WeightNoiseType.ADDITIVE_CONSTANT
    • WeightNoiseType.NONE
    • WeightNoiseType.PCM_READ
  • WeightRemapType
    • WeightRemapType.CHANNELWISE_SYMMETRIC
    • WeightRemapType.LAYERWISE_SYMMETRIC
    • WeightRemapType.NONE
• aihwkit.simulator.parameters.helpers module
  • get_bindings_class()
  • parameters_to_bindings()
  • tile_parameters_to_bindings()
• aihwkit.simulator.parameters.inference module
  • DriftParameter
    • DriftParameter.g_offset
    • DriftParameter.log_g0
    • DriftParameter.nu_dtod
    • DriftParameter.nu_k
    • DriftParameter.nu_std
    • DriftParameter.w_noise_std
    • DriftParameter.w_offset
    • DriftParameter.wg_ratio
  • SimpleDriftParameter
    • SimpleDriftParameter.bindings_class
    • SimpleDriftParameter.bindings_module
    • SimpleDriftParameter.nu
    • SimpleDriftParameter.reset_tol
    • SimpleDriftParameter.t_0
  • WeightClipParameter
    • WeightClipParameter.bindings_class
    • WeightClipParameter.fixed_value
    • WeightClipParameter.sigma
    • WeightClipParameter.type
  • WeightModifierParameter
    • WeightModifierParameter.assumed_wmax
    • WeightModifierParameter.bindings_class
    • WeightModifierParameter.bindings_module
    • WeightModifierParameter.coeffs
    • WeightModifierParameter.copy_last_column
    • WeightModifierParameter.dorefa_clip
    • WeightModifierParameter.enable_during_test
    • WeightModifierParameter.g_max
    • WeightModifierParameter.pcm_prob_at_gmax
    • WeightModifierParameter.pcm_prob_at_random
    • WeightModifierParameter.pcm_prob_at_reset
    • WeightModifierParameter.pcm_t0
    • WeightModifierParameter.pcm_t_inference
    • WeightModifierParameter.pcm_zero_thres
    • WeightModifierParameter.pdrop
    • WeightModifierParameter.per_batch_sample
    • WeightModifierParameter.rel_to_actual_wmax
    • WeightModifierParameter.res
    • WeightModifierParameter.std_dev
    • WeightModifierParameter.sto_round
    • WeightModifierParameter.type
  • WeightRemapParameter
    • WeightRemapParameter.bindings_class
    • WeightRemapParameter.max_scale_range
    • WeightRemapParameter.max_scale_ref
    • WeightRemapParameter.remapped_wmax
    • WeightRemapParameter.type
• aihwkit.simulator.parameters.mapping module
  • MappableRPU
    • MappableRPU.get_default_tile_module_class()
    • MappableRPU.mapping
    • MappableRPU.tile_array_class
  • MappingParameter
    • MappingParameter.compatible_with()
    • MappingParameter.digital_bias
    • MappingParameter.learn_out_scaling
    • MappingParameter.max_input_size
    • MappingParameter.max_output_size
    • MappingParameter.out_scaling_columnwise
    • MappingParameter.weight_scaling_columnwise
    • MappingParameter.weight_scaling_lr_compensation
    • MappingParameter.weight_scaling_omega
• aihwkit.simulator.parameters.pre_post module
  • InputRangeParameter
    • InputRangeParameter.calibration_info
    • InputRangeParameter.decay
    • InputRangeParameter.enable
    • InputRangeParameter.gradient_relative
    • InputRangeParameter.gradient_scale
    • InputRangeParameter.init_from_data
    • InputRangeParameter.init_std_alpha
    • InputRangeParameter.init_value
    • InputRangeParameter.input_min_percentage
    • InputRangeParameter.learn_input_range
    • InputRangeParameter.manage_output_clipping
    • InputRangeParameter.output_min_percentage
    • InputRangeParameter.supports_manage_output_clipping()
  • PrePostProcessingParameter
    • PrePostProcessingParameter.input_range
  • PrePostProcessingRPU
    • PrePostProcessingRPU.pre_post
• aihwkit.simulator.parameters.runtime module
  • RuntimeParameter
    • RuntimeParameter.data_type
    • RuntimeParameter.offload_gradient
    • RuntimeParameter.offload_input
    • RuntimeParameter.on_the_fly_bindings
• aihwkit.simulator.parameters.training module
  • UpdateParameters
    • UpdateParameters.bindings_class
    • UpdateParameters.bindings_module
    • UpdateParameters.d_res_implicit
    • UpdateParameters.d_sparsity
    • UpdateParameters.desired_bl
    • UpdateParameters.fixed_bl
    • UpdateParameters.pulse_type
    • UpdateParameters.res
    • UpdateParameters.sto_round
    • UpdateParameters.um_grad_scale
    • UpdateParameters.update_bl_management
    • UpdateParameters.update_management
    • UpdateParameters.x_res_implicit