aihwkit.utils package

Utilities and helpers for aihwkit.

Submodules

• aihwkit.utils.analog_info module
  • AnalogInfo
    • AnalogInfo.calculate_num_analog()
    • AnalogInfo.calculate_num_tiles()
    • AnalogInfo.create_layer_summary()
    • AnalogInfo.register_hooks_recursively()
  • LayerInfo
    • LayerInfo.calculate_reuse_factor()
    • LayerInfo.input_size
    • LayerInfo.isanalog
    • LayerInfo.kernel_size
    • LayerInfo.layer_summary_dict()
    • LayerInfo.module
    • LayerInfo.name
    • LayerInfo.num_tiles
    • LayerInfo.output_size
    • LayerInfo.reuse_factor
    • LayerInfo.set_kernel_size()
    • LayerInfo.set_tiles_info()
    • LayerInfo.tiles_info
    • LayerInfo.tiles_summary_dict()
  • TileInfo
    • TileInfo.log_in_size
    • TileInfo.log_out_size
    • TileInfo.phy_in_size
    • TileInfo.phy_out_size
    • TileInfo.tile_summary_dict()
    • TileInfo.utilization
  • analog_summary()
• aihwkit.utils.export module
  • fusion_export()
  • fusion_import()
• aihwkit.utils.fitting module
  • fit_measurements()
  • model_response()
• aihwkit.utils.legacy module
  • convert_legacy_checkpoint()
• aihwkit.utils.visualization module
  • compute_pulse_response()
  • compute_pulse_statistics()
  • estimate_n_steps()
  • get_tile_for_plotting()
  • plot_device()
  • plot_device_compact()
  • plot_device_symmetry()
  • plot_model_programming_error()
  • plot_programming_error()
  • plot_pulse_response()
  • plot_pulse_statistics()
  • plot_response_overview()
  • plot_weight_drift()
• aihwkit.utils.visualization_web module
  • plot_device_compact_web()
  • save_plots_for_web()
  • set_dark_style()