File: jit.rst

package info (click to toggle)
pytorch 2.9.1%2Bdfsg-1~exp2
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 180,096 kB
  • sloc: python: 1,473,255; cpp: 942,030; ansic: 79,796; asm: 7,754; javascript: 2,502; java: 1,962; sh: 1,809; makefile: 628; xml: 8
file content (59 lines) | stat: -rw-r--r-- 1,226 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
TorchScript
===========

.. toctree::
    :maxdepth: 1
    :hidden:

    jit_language_reference
    jit_language_reference_v2
    jit_python_reference
    jit_unsupported
    torch.jit.supported_ops <jit_builtin_functions>

.. warning::
    TorchScript is deprecated, please use
    `torch.export <https://docs.pytorch.org/docs/stable/export.html>`__ instead.

.. automodule:: torch.jit

Creating TorchScript Code
--------------------------

.. autosummary::
    :toctree: generated
    :nosignatures:

    script
    trace
    script_if_tracing
    trace_module
    fork
    wait
    ScriptModule
    ScriptFunction
    freeze
    optimize_for_inference
    enable_onednn_fusion
    onednn_fusion_enabled
    set_fusion_strategy
    strict_fusion
    save
    load
    ignore
    unused
    interface
    isinstance
    Attribute
    annotate


.. This package is missing doc. Adding it here for coverage
.. This does not add anything to the rendered page.
.. py:module:: torch.jit.supported_ops
.. py:module:: torch.jit.unsupported_tensor_ops
.. py:module:: torch.jit.mobile
.. py:module:: torch.jit.annotations
.. py:module:: torch.jit.frontend
.. py:module:: torch.jit.generate_bytecode
.. py:module:: torch.jit.quantized