aihwkit.simulator.configs package
Configurations for resistive processing units.
Submodules
- aihwkit.simulator.configs.compounds module
BufferedTransferCompound
ChoppedTransferCompound
ChoppedTransferCompound.as_bindings()
ChoppedTransferCompound.auto_granularity
ChoppedTransferCompound.auto_momentum
ChoppedTransferCompound.auto_scale
ChoppedTransferCompound.bindings_class
ChoppedTransferCompound.buffer_granularity
ChoppedTransferCompound.correct_gradient_magnitudes
ChoppedTransferCompound.fast_lr
ChoppedTransferCompound.forget_buffer
ChoppedTransferCompound.in_chop_prob
ChoppedTransferCompound.in_chop_random
ChoppedTransferCompound.momentum
ChoppedTransferCompound.no_buffer
ChoppedTransferCompound.out_chop_prob
ChoppedTransferCompound.scale_transfer_lr
ChoppedTransferCompound.step
ChoppedTransferCompound.transfer_columns
ChoppedTransferCompound.transfer_forward
ChoppedTransferCompound.transfer_lr
ChoppedTransferCompound.transfer_update
ChoppedTransferCompound.units_in_mbatch
DifferenceUnitCell
DigitalRankUpdateCell
DynamicTransferCompound
DynamicTransferCompound.as_bindings()
DynamicTransferCompound.bindings_class
DynamicTransferCompound.buffer_cap
DynamicTransferCompound.buffer_granularity
DynamicTransferCompound.experimental_correct_accumulation
DynamicTransferCompound.fast_lr
DynamicTransferCompound.forget_buffer
DynamicTransferCompound.in_chop_prob
DynamicTransferCompound.in_chop_random
DynamicTransferCompound.momentum
DynamicTransferCompound.scale_transfer_lr
DynamicTransferCompound.step
DynamicTransferCompound.tail_weightening
DynamicTransferCompound.transfer_columns
DynamicTransferCompound.transfer_forward
DynamicTransferCompound.transfer_lr
DynamicTransferCompound.transfer_update
MixedPrecisionCompound
MixedPrecisionCompound.as_bindings()
MixedPrecisionCompound.bindings_class
MixedPrecisionCompound.granularity
MixedPrecisionCompound.n_d_bins
MixedPrecisionCompound.n_rows_per_transfer
MixedPrecisionCompound.n_x_bins
MixedPrecisionCompound.random_row
MixedPrecisionCompound.stoc_round_d
MixedPrecisionCompound.stoc_round_x
MixedPrecisionCompound.transfer_every
MixedPrecisionCompound.transfer_lr
OneSidedUnitCell
ReferenceUnitCell
TransferCompound
TransferCompound.as_bindings()
TransferCompound.bindings_class
TransferCompound.fast_lr
TransferCompound.gamma
TransferCompound.gamma_vec
TransferCompound.n_reads_per_transfer
TransferCompound.no_self_transfer
TransferCompound.random_selection
TransferCompound.scale_transfer_lr
TransferCompound.transfer_columns
TransferCompound.transfer_every
TransferCompound.transfer_every_vec
TransferCompound.transfer_forward
TransferCompound.transfer_lr
TransferCompound.transfer_lr_vec
TransferCompound.transfer_update
TransferCompound.units_in_mbatch
TransferCompound.with_reset_prob
UnitCell
VectorUnitCell
- aihwkit.simulator.configs.configs module
DigitalRankUpdateRPUConfig
FloatingPointRPUConfig
IOManagedRPUConfig
InferenceRPUConfig
InferenceRPUConfig.backward
InferenceRPUConfig.clip
InferenceRPUConfig.compatible_with()
InferenceRPUConfig.device
InferenceRPUConfig.drift_compensation
InferenceRPUConfig.forward
InferenceRPUConfig.modifier
InferenceRPUConfig.noise_model
InferenceRPUConfig.remap
InferenceRPUConfig.tile_array_class
InferenceRPUConfig.tile_class
InferenceRPUConfig.update
SingleRPUConfig
TorchInferenceRPUConfig
TorchInferenceRPUConfigIRDropT
UnitCellRPUConfig
- aihwkit.simulator.configs.devices module
ConstantStepDevice
ExpStepDevice
FloatingPointDevice
IdealDevice
LinearStepDevice
LinearStepDevice.allow_increasing
LinearStepDevice.apply_write_noise_on_set
LinearStepDevice.bindings_class
LinearStepDevice.gamma_down
LinearStepDevice.gamma_down_dtod
LinearStepDevice.gamma_up
LinearStepDevice.gamma_up_dtod
LinearStepDevice.mean_bound_reference
LinearStepDevice.mult_noise
LinearStepDevice.reverse_down
LinearStepDevice.reverse_offset
LinearStepDevice.reverse_up
LinearStepDevice.write_noise_std
PiecewiseStepDevice
PowStepDevice
PowStepReferenceDevice
PowStepReferenceDevice.bindings_class
PowStepReferenceDevice.n_estimation_steps
PowStepReferenceDevice.pow_gamma
PowStepReferenceDevice.pow_gamma_dtod
PowStepReferenceDevice.pow_up_down
PowStepReferenceDevice.pow_up_down_dtod
PowStepReferenceDevice.reference_mean
PowStepReferenceDevice.reference_std
PowStepReferenceDevice.subtract_symmetry_point
PulsedDevice
PulsedDevice.as_bindings()
PulsedDevice.bindings_class
PulsedDevice.bindings_module
PulsedDevice.construction_seed
PulsedDevice.corrupt_devices_prob
PulsedDevice.corrupt_devices_range
PulsedDevice.count_pulses
PulsedDevice.diffusion
PulsedDevice.diffusion_dtod
PulsedDevice.drift
PulsedDevice.dw_min
PulsedDevice.dw_min_dtod
PulsedDevice.dw_min_dtod_log_normal
PulsedDevice.dw_min_std
PulsedDevice.enforce_consistency
PulsedDevice.lifetime
PulsedDevice.lifetime_dtod
PulsedDevice.perfect_bias
PulsedDevice.requires_decay()
PulsedDevice.requires_diffusion()
PulsedDevice.reset
PulsedDevice.reset_dtod
PulsedDevice.reset_std
PulsedDevice.up_down
PulsedDevice.up_down_dtod
PulsedDevice.w_max
PulsedDevice.w_max_dtod
PulsedDevice.w_min
PulsedDevice.w_min_dtod
SoftBoundsDevice
SoftBoundsPmaxDevice
SoftBoundsReferenceDevice
SoftBoundsReferenceDevice.apply_write_noise_on_set
SoftBoundsReferenceDevice.bindings_class
SoftBoundsReferenceDevice.mult_noise
SoftBoundsReferenceDevice.reference_mean
SoftBoundsReferenceDevice.reference_std
SoftBoundsReferenceDevice.slope_down_dtod
SoftBoundsReferenceDevice.slope_up_dtod
SoftBoundsReferenceDevice.subtract_symmetry_point
SoftBoundsReferenceDevice.write_noise_std
- aihwkit.simulator.configs.helpers module
- aihwkit.simulator.configs.utils module