|
10 | 10 |
|
11 | 11 | <meta name="viewport" content="width=device-width, initial-scale=1.0"> |
12 | 12 |
|
13 | | - <title>torch._dynamo — PyTorch 2.1 documentation</title> |
| 13 | + <title>torch._dynamo — PyTorch master documentation</title> |
14 | 14 |
|
15 | 15 |
|
16 | 16 |
|
|
44 | 44 | <link rel="next" title="torch.fft" href="fft.html" /> |
45 | 45 | <link rel="prev" title="Probability distributions - torch.distributions" href="distributions.html" /> |
46 | 46 |
|
| 47 | +<!-- |
| 48 | + Search engines should not index the master version of documentation. |
| 49 | + Stable documentation are built without release == 'master'. |
| 50 | +--> |
| 51 | +<meta name="robots" content="noindex"> |
| 52 | + |
47 | 53 |
|
48 | 54 | <!-- Google Analytics --> |
49 | 55 |
|
|
231 | 237 | <div class="pytorch-left-menu-search"> |
232 | 238 |
|
233 | 239 | <div class="version"> |
234 | | - <a href='https://pytorch.org/docs/versions.html'>2.1 ▼</a> |
| 240 | + <a href='https://pytorch.org/docs/versions.html'>master (2.1.0a0+gitbafa2c4 ) ▼</a> |
235 | 241 | </div> |
236 | 242 |
|
237 | 243 |
|
|
251 | 257 |
|
252 | 258 |
|
253 | 259 |
|
| 260 | +<div> |
| 261 | + <a style="color:#F05732" href="https://pytorch.org/docs/stable/_dynamo.html"> |
| 262 | + You are viewing unstable developer preview docs. |
| 263 | + Click here to view docs for latest stable release. |
| 264 | + </a> |
| 265 | +</div> |
| 266 | + |
254 | 267 |
|
255 | 268 |
|
256 | 269 |
|
|
589 | 602 |
|
590 | 603 | <dl class="py function"> |
591 | 604 | <dt class="sig sig-object py" id="torch._dynamo.export"> |
592 | | -<span class="sig-prename descclassname"><span class="pre">torch._dynamo.</span></span><span class="sig-name descname"><span class="pre">export</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">f</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">aten_graph</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">decomposition_table</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tracing_mode</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'symbolic'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">constraints</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">assume_static_by_default</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">kwargs</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/torch/_dynamo/eval_frame.html#export"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torch._dynamo.export" title="Permalink to this definition">¶</a></dt> |
| 605 | +<span class="sig-prename descclassname"><span class="pre">torch._dynamo.</span></span><span class="sig-name descname"><span class="pre">export</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">f</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">aten_graph</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">pre_autograd</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">decomposition_table</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tracing_mode</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'symbolic'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">constraints</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">assume_static_by_default</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">kwargs</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/torch/_dynamo/eval_frame.html#export"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torch._dynamo.export" title="Permalink to this definition">¶</a></dt> |
593 | 606 | <dd><p>Export an input function f to a format that can be executed outside of PyTorch using the FX graph.</p> |
594 | 607 | <dl class="field-list simple"> |
595 | 608 | <dt class="field-odd">Parameters<span class="colon">:</span></dt> |
|
598 | 611 | <li><p><strong>*args</strong> – Variable length argument list to be passed to the function f.</p></li> |
599 | 612 | <li><p><strong>aten_graph</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.11)"><em>bool</em></a>) – If True, exports a graph with ATen operators.</p></li> |
600 | 613 | <li><p><strong>False</strong> (<em>If</em>) – </p></li> |
601 | | -<li><p><strong>False.</strong> (<em>exports a graph with Python operators. Default is</em>) – </p></li> |
| 614 | +<li><p><strong>False.</strong> (<em>Default is</em>) – </p></li> |
| 615 | +<li><p><strong>pre_autograd</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.11)"><em>bool</em></a>) – If True, exports a graph with ATen operators,</p></li> |
| 616 | +<li><p><strong>tranformations</strong> (<em>but before autograd has run. This can be useful if you want to apply further</em>) – </p></li> |
| 617 | +<li><p><strong>autograd.</strong> (<em>on a graph before running it through</em>) – </p></li> |
| 618 | +<li><p><strong>set.</strong> (<em>This flag is only valid if aten_graph=True is</em>) – </p></li> |
| 619 | +<li><p><strong>False.</strong> – </p></li> |
602 | 620 | <li><p><strong>decomposition_table</strong> (<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#dict" title="(in Python v3.11)"><em>dict</em></a>) – A dictionary that maps operators to their decomposition functions.</p></li> |
603 | 621 | <li><p><strong>None.</strong> (<em>Required if aten_graph</em><em> or </em><em>tracing_mode is specified. Default is</em>) – </p></li> |
604 | 622 | <li><p><strong>tracing_mode</strong> (<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.11)"><em>str</em></a>) – Specifies the tracing mode. Must be set to “real” if decomposition_table is not specified.</p></li> |
|
0 commit comments