aihwkit.simulator.rpu_base module

Bindings for the RPU simulator.

Submodules

• aihwkit.simulator.rpu_base.tiles
  • AnalogTile
    • AnalogTile.get_meta_parameters()
  • FloatingPointTile
    • FloatingPointTile.backward()
    • FloatingPointTile.backward_indexed()
    • FloatingPointTile.clip_weights()
    • FloatingPointTile.decay_weights()
    • FloatingPointTile.diffuse_weights()
    • FloatingPointTile.drift_weights()
    • FloatingPointTile.dump_extra()
    • FloatingPointTile.forward()
    • FloatingPointTile.forward_indexed()
    • FloatingPointTile.get_brief_info()
    • FloatingPointTile.get_d_size()
    • FloatingPointTile.get_hidden_parameter_names()
    • FloatingPointTile.get_hidden_parameters()
    • FloatingPointTile.get_hidden_update_index()
    • FloatingPointTile.get_info()
    • FloatingPointTile.get_learning_rate()
    • FloatingPointTile.get_meta_parameters()
    • FloatingPointTile.get_pulse_counters()
    • FloatingPointTile.get_shared_weights_if()
    • FloatingPointTile.get_weights()
    • FloatingPointTile.get_x_size()
    • FloatingPointTile.has_matrix_indices()
    • FloatingPointTile.load_extra()
    • FloatingPointTile.modify_weights()
    • FloatingPointTile.remap_weights()
    • FloatingPointTile.reset_columns()
    • FloatingPointTile.reset_delta_weights()
    • FloatingPointTile.set_delta_weights()
    • FloatingPointTile.set_hidden_parameters()
    • FloatingPointTile.set_hidden_update_index()
    • FloatingPointTile.set_learning_rate()
    • FloatingPointTile.set_matrix_indices()
    • FloatingPointTile.set_shared_weights()
    • FloatingPointTile.set_verbosity_level()
    • FloatingPointTile.set_weights()
    • FloatingPointTile.set_weights_uniform_random()
    • FloatingPointTile.update()
    • FloatingPointTile.update_indexed()
  • WeightClipParameter
    • WeightClipParameter.fixed_value
    • WeightClipParameter.sigma
    • WeightClipParameter.type
  • WeightClipType
    • WeightClipType.AverageChannelMax
    • WeightClipType.FixedValue
    • WeightClipType.LayerGaussian
    • WeightClipType.None
    • WeightClipType.value
  • WeightModifierParameter
    • WeightModifierParameter.assumed_wmax
    • 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
  • WeightModifierType
    • WeightModifierType.AddNormal
    • WeightModifierType.Copy
    • WeightModifierType.Discretize
    • WeightModifierType.DiscretizeAddNormal
    • WeightModifierType.DoReFa
    • WeightModifierType.DropConnect
    • WeightModifierType.MultNormal
    • WeightModifierType.None
    • WeightModifierType.PCMNoise
    • WeightModifierType.Poly
    • WeightModifierType.ProgNoise
    • WeightModifierType.value
  • WeightRemapParameter
    • WeightRemapParameter.max_scale_range
    • WeightRemapParameter.max_scale_ref
    • WeightRemapParameter.remapped_wmax
    • WeightRemapParameter.type
  • WeightRemapType
    • WeightRemapType.ChannelwiseSymmetric
    • WeightRemapType.LayerwiseSymmetric
    • WeightRemapType.None
    • WeightRemapType.value