aihwkit.simulator.tiles package

High level analog tiles.

Submodules

• aihwkit.simulator.tiles.analog module
  • AnalogTile
    • AnalogTile.forward()
    • AnalogTile.supports_ddp
  • AnalogTileWithoutPeriphery
    • AnalogTileWithoutPeriphery.backward()
    • AnalogTileWithoutPeriphery.forward()
    • AnalogTileWithoutPeriphery.get_learning_rate()
    • AnalogTileWithoutPeriphery.joint_forward()
    • AnalogTileWithoutPeriphery.set_learning_rate()
    • AnalogTileWithoutPeriphery.supports_indexed
    • AnalogTileWithoutPeriphery.update()
• aihwkit.simulator.tiles.array module
  • QuantizedTileModuleArray
    • QuantizedTileModuleArray.forward()
  • TileModuleArray
    • TileModuleArray.forward()
    • TileModuleArray.get_split_sizes()
    • TileModuleArray.get_weights()
    • TileModuleArray.set_weights()
    • TileModuleArray.supports_indexed
• aihwkit.simulator.tiles.base module
  • AnalogTileStateNames
    • AnalogTileStateNames.ANALOG_STATE_NAME
    • AnalogTileStateNames.ANALOG_STATE_PREFIX
    • AnalogTileStateNames.CLASS
    • AnalogTileStateNames.CONTEXT
    • AnalogTileStateNames.EXTRA
    • AnalogTileStateNames.HIDDEN_PARAMETERS
    • AnalogTileStateNames.HIDDEN_PARAMETER_NAMES
    • AnalogTileStateNames.LR
    • AnalogTileStateNames.MAPPING_SCALES
    • AnalogTileStateNames.OUT_SCALING
    • AnalogTileStateNames.RPU_CONFIG
    • AnalogTileStateNames.SHARED_WEIGHTS
    • AnalogTileStateNames.VERSION
    • AnalogTileStateNames.WEIGHTS
    • AnalogTileStateNames.get_field_names()
  • BaseTile
    • BaseTile.backward()
    • BaseTile.joint_forward()
    • BaseTile.update()
  • SimulatorTile
    • SimulatorTile.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()
  • SimulatorTileWrapper
    • SimulatorTileWrapper.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
  • CustomRPUConfig
    • CustomRPUConfig.backward
    • CustomRPUConfig.forward
    • CustomRPUConfig.simulator_tile_class
    • CustomRPUConfig.tile_array_class
    • CustomRPUConfig.tile_class
    • CustomRPUConfig.update
  • CustomSimulatorTile
    • CustomSimulatorTile.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()
  • CustomTile
    • CustomTile.forward()
    • CustomTile.post_update_step()
    • CustomTile.supports_indexed
  • CustomUpdateParameters
    • CustomUpdateParameters.gradient_noise
• aihwkit.simulator.tiles.floating_point module
  • FloatingPointTile
    • FloatingPointTile.forward()
• aihwkit.simulator.tiles.functions module
  • AnalogFunction
    • AnalogFunction.backward()
    • AnalogFunction.forward()
• aihwkit.simulator.tiles.inference module
  • InferenceTile
    • InferenceTile.forward()
    • InferenceTile.supports_ddp
  • InferenceTileWithPeriphery
    • InferenceTileWithPeriphery.cpu()
    • InferenceTileWithPeriphery.cuda()
    • InferenceTileWithPeriphery.drift_weights()
    • InferenceTileWithPeriphery.init_mapping_scales()
    • InferenceTileWithPeriphery.post_forward()
    • InferenceTileWithPeriphery.post_update_step()
    • InferenceTileWithPeriphery.program_weights()
• aihwkit.simulator.tiles.inference_torch module
  • InputRangeForward
    • InputRangeForward.backward()
    • InputRangeForward.forward()
  • TorchInferenceTile
    • TorchInferenceTile.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_ddp
    • TorchInferenceTile.supports_indexed
• aihwkit.simulator.tiles.module module
  • TileModule
    • TileModule.compatible_with()
    • TileModule.cpu()
    • TileModule.cuda()
    • TileModule.extra_repr()
    • TileModule.get_analog_state_name()
    • TileModule.is_floating_point()
    • TileModule.replace_with()
    • TileModule.set_load_rpu_config_state()
    • TileModule.state_dict()
    • TileModule.supports_ddp
    • TileModule.to()
• aihwkit.simulator.tiles.periphery module
  • TileWithPeriphery
    • TileWithPeriphery.add_quant_periphery_bias()
    • TileWithPeriphery.apply_input_range()
    • TileWithPeriphery.apply_out_scaling()
    • TileWithPeriphery.apply_quant_periphery_scales()
    • 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.init_periph_quant_processing()
    • 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_indexed
    • TileWithPeriphery.update()
    • TileWithPeriphery.update_indexed()
• aihwkit.simulator.tiles.rpucuda module
  • RPUCudaSimulatorTileWrapper
    • RPUCudaSimulatorTileWrapper.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
  • TorchSimulatorTile
    • TorchSimulatorTile.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
  • TorchTransferTile
    • TorchTransferTile.forward()
    • TorchTransferTile.post_update_step()
    • TorchTransferTile.replace_with()
    • TorchTransferTile.supports_ddp
    • TorchTransferTile.supports_indexed
  • TransferSimulatorTile
    • TransferSimulatorTile.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
  • UniformQuantize
    • UniformQuantize.backward()
    • UniformQuantize.forward()
  • isinf()