.. # Copyright (c) 2023 Graphcore Ltd. All rights reserved. # Copyright (c) 2007-2023 by the Sphinx team. All rights reserved. unit\_scaling.transforms ======================== .. automodule:: unit_scaling.transforms .. rubric:: Functions .. autosummary:: :toctree: compile prune_non_float_tensors prune_same_scale_tensors prune_selected_nodes simulate_format simulate_fp8 track_scales unit_scale .. rubric:: Classes .. autosummary:: :toctree: :template: custom-class-template.rst Metrics