Back to Pytorch

JIT Utils - torch.utils.jit

docs/source/jit_utils.md

2.12.0388 B
Original Source

JIT Utils - torch.utils.jit

{eval-rst}
.. automodule:: torch.utils.jit

torch.utils.jit.log_extract

{eval-rst}
.. automodule:: torch.utils.jit.log_extract
{eval-rst}
.. currentmodule:: torch.utils.jit.log_extract
{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:

    run_baseline_no_fusion
    run_nnc
    run_nvfuser
    run_test