aihwkit.simulator.tiles package
High level analog tiles.
Submodules
- aihwkit.simulator.tiles.analog module
- aihwkit.simulator.tiles.array module
- aihwkit.simulator.tiles.base module
AnalogTileStateNamesAnalogTileStateNames.ANALOG_STATE_NAMEAnalogTileStateNames.ANALOG_STATE_PREFIXAnalogTileStateNames.CLASSAnalogTileStateNames.CONTEXTAnalogTileStateNames.EXTRAAnalogTileStateNames.HIDDEN_PARAMETERSAnalogTileStateNames.HIDDEN_PARAMETER_NAMESAnalogTileStateNames.LRAnalogTileStateNames.MAPPING_SCALESAnalogTileStateNames.OUT_SCALINGAnalogTileStateNames.RPU_CONFIGAnalogTileStateNames.SHARED_WEIGHTSAnalogTileStateNames.VERSIONAnalogTileStateNames.WEIGHTSAnalogTileStateNames.get_field_names()
BaseTileSimulatorTileSimulatorTile.backward()SimulatorTile.dump_extra()SimulatorTile.forward()SimulatorTile.get_brief_info()SimulatorTile.get_d_size()SimulatorTile.get_hidden_parameter_names()SimulatorTile.get_hidden_parameters()SimulatorTile.get_learning_rate()SimulatorTile.get_meta_parameters()SimulatorTile.get_weights()SimulatorTile.get_x_size()SimulatorTile.load_extra()SimulatorTile.set_hidden_parameters()SimulatorTile.set_learning_rate()SimulatorTile.set_weights()SimulatorTile.set_weights_uniform_random()SimulatorTile.update()
SimulatorTileWrapperSimulatorTileWrapper.cpu()SimulatorTileWrapper.cuda()SimulatorTileWrapper.get_analog_ctx()SimulatorTileWrapper.get_analog_state()SimulatorTileWrapper.get_brief_info()SimulatorTileWrapper.get_data_type()SimulatorTileWrapper.get_dtype()SimulatorTileWrapper.get_forward_out_bound()SimulatorTileWrapper.get_hidden_parameters()SimulatorTileWrapper.get_learning_rate()SimulatorTileWrapper.get_runtime()SimulatorTileWrapper.get_tensor_view()SimulatorTileWrapper.post_update_step()SimulatorTileWrapper.set_hidden_parameters()SimulatorTileWrapper.set_learning_rate()SimulatorTileWrapper.set_verbosity_level()SimulatorTileWrapper.update()SimulatorTileWrapper.update_indexed()
TileModuleBase
- aihwkit.simulator.tiles.custom module
CustomRPUConfigCustomSimulatorTileCustomSimulatorTile.backward()CustomSimulatorTile.extra_repr()CustomSimulatorTile.forward()CustomSimulatorTile.get_brief_info()CustomSimulatorTile.get_d_size()CustomSimulatorTile.get_learning_rate()CustomSimulatorTile.get_meta_parameters()CustomSimulatorTile.get_weights()CustomSimulatorTile.get_x_size()CustomSimulatorTile.set_config()CustomSimulatorTile.set_learning_rate()CustomSimulatorTile.set_weights()CustomSimulatorTile.set_weights_uniform_random()CustomSimulatorTile.update()
CustomTileCustomUpdateParameters
- aihwkit.simulator.tiles.floating_point module
- aihwkit.simulator.tiles.functions module
- aihwkit.simulator.tiles.inference module
- aihwkit.simulator.tiles.inference_torch module
InputRangeForwardTorchInferenceTileTorchInferenceTile.forward()TorchInferenceTile.get_forward_parameters()TorchInferenceTile.init_input_processing()TorchInferenceTile.post_update_step()TorchInferenceTile.pre_forward()TorchInferenceTile.set_forward_parameters()TorchInferenceTile.set_scales()TorchInferenceTile.supports_ddpTorchInferenceTile.supports_indexed
- aihwkit.simulator.tiles.module module
- aihwkit.simulator.tiles.periphery module
TileWithPeripheryTileWithPeriphery.apply_input_range()TileWithPeriphery.apply_out_scaling()TileWithPeriphery.apply_weight_scaling()TileWithPeriphery.backward()TileWithPeriphery.backward_indexed()TileWithPeriphery.cpu()TileWithPeriphery.cuda()TileWithPeriphery.get_learned_out_scales()TileWithPeriphery.get_learning_rate()TileWithPeriphery.get_mapping_scales()TileWithPeriphery.get_scales()TileWithPeriphery.get_weights()TileWithPeriphery.init_input_processing()TileWithPeriphery.init_learned_out_scales()TileWithPeriphery.init_mapping_scales()TileWithPeriphery.is_indexed()TileWithPeriphery.joint_forward()TileWithPeriphery.joint_forward_indexed()TileWithPeriphery.post_backward()TileWithPeriphery.post_forward()TileWithPeriphery.pre_backward()TileWithPeriphery.pre_forward()TileWithPeriphery.pre_update()TileWithPeriphery.program_weights()TileWithPeriphery.read_weights()TileWithPeriphery.remap_weights()TileWithPeriphery.set_indexed()TileWithPeriphery.set_input_range()TileWithPeriphery.set_learned_out_scales()TileWithPeriphery.set_learning_rate()TileWithPeriphery.set_mapping_scales()TileWithPeriphery.set_scales()TileWithPeriphery.set_weights()TileWithPeriphery.supports_indexedTileWithPeriphery.update()TileWithPeriphery.update_indexed()
- aihwkit.simulator.tiles.rpucuda module
RPUCudaSimulatorTileWrapperRPUCudaSimulatorTileWrapper.cpu()RPUCudaSimulatorTileWrapper.cuda()RPUCudaSimulatorTileWrapper.decay_weights()RPUCudaSimulatorTileWrapper.diffuse_weights()RPUCudaSimulatorTileWrapper.drift_weights()RPUCudaSimulatorTileWrapper.dump_extra()RPUCudaSimulatorTileWrapper.ensure_shared_weights()RPUCudaSimulatorTileWrapper.get_backward_parameters()RPUCudaSimulatorTileWrapper.get_forward_out_bound()RPUCudaSimulatorTileWrapper.get_forward_parameters()RPUCudaSimulatorTileWrapper.get_hidden_update_index()RPUCudaSimulatorTileWrapper.load_extra()RPUCudaSimulatorTileWrapper.post_update_step()RPUCudaSimulatorTileWrapper.reset()RPUCudaSimulatorTileWrapper.reset_columns()RPUCudaSimulatorTileWrapper.reset_delta_weights()RPUCudaSimulatorTileWrapper.set_backward_parameters()RPUCudaSimulatorTileWrapper.set_delta_weights()RPUCudaSimulatorTileWrapper.set_forward_parameters()RPUCudaSimulatorTileWrapper.set_hidden_update_index()RPUCudaSimulatorTileWrapper.set_verbosity_level()
- aihwkit.simulator.tiles.torch_tile module
TorchSimulatorTileTorchSimulatorTile.check_rpu_config_support()TorchSimulatorTile.clip_weights()TorchSimulatorTile.extra_repr()TorchSimulatorTile.forward()TorchSimulatorTile.get_brief_info()TorchSimulatorTile.get_d_size()TorchSimulatorTile.get_weights()TorchSimulatorTile.get_x_size()TorchSimulatorTile.modify_weight()TorchSimulatorTile.register_weight_hook()TorchSimulatorTile.remap_weights()TorchSimulatorTile.set_config()TorchSimulatorTile.set_weights()
- aihwkit.simulator.tiles.transfer module
TorchTransferTileTransferSimulatorTileTransferSimulatorTile.backward()TransferSimulatorTile.dump_extra()TransferSimulatorTile.forward()TransferSimulatorTile.get_brief_info()TransferSimulatorTile.get_d_size()TransferSimulatorTile.get_hidden_parameter_names()TransferSimulatorTile.get_hidden_parameters()TransferSimulatorTile.get_learning_rate()TransferSimulatorTile.get_meta_parameters()TransferSimulatorTile.get_weights()TransferSimulatorTile.get_x_size()TransferSimulatorTile.load_extra()TransferSimulatorTile.post_update_step()TransferSimulatorTile.set_hidden_parameters()TransferSimulatorTile.set_learning_rate()TransferSimulatorTile.set_weights()TransferSimulatorTile.set_weights_uniform_random()TransferSimulatorTile.update()
- aihwkit.simulator.tiles.utils module