Generate Python docs from pytorch/pytorch@0a26e5f

Author: pytorchbot

docs/main/_images/RReLU.png

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/index.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 
@@ -527,7 +527,7 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
     <span class="s1">&#39;set_float32_matmul_precision&#39;</span><span class="p">,</span> <span class="s1">&#39;get_float32_matmul_precision&#39;</span><span class="p">,</span>
     <span class="s1">&#39;set_warn_always&#39;</span><span class="p">,</span> <span class="s1">&#39;is_warn_always_enabled&#39;</span><span class="p">,</span> <span class="s1">&#39;SymInt&#39;</span><span class="p">,</span> <span class="s1">&#39;SymFloat&#39;</span><span class="p">,</span>
     <span class="s1">&#39;SymBool&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_not&#39;</span><span class="p">,</span> <span class="s1">&#39;unravel_index&#39;</span><span class="p">,</span>
-    <span class="s1">&#39;sym_int&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_float&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_max&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_min&#39;</span><span class="p">,</span> <span class="s1">&#39;compile&#39;</span><span class="p">,</span> <span class="s1">&#39;vmap&#39;</span><span class="p">,</span>
+    <span class="s1">&#39;sym_int&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_float&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_max&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_min&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_ite&#39;</span><span class="p">,</span> <span class="s1">&#39;compile&#39;</span><span class="p">,</span> <span class="s1">&#39;vmap&#39;</span><span class="p">,</span>
     <span class="s1">&#39;export&#39;</span><span class="p">,</span> <span class="s1">&#39;autocast&#39;</span><span class="p">,</span> <span class="s1">&#39;cond&#39;</span><span class="p">,</span>
 <span class="p">]</span>
 
@@ -862,6 +862,9 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
     <span class="k">def</span> <span class="nf">__sym_not__</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="s2">&quot;SymBool&quot;</span><span class="p">:</span>
         <span class="k">raise</span> <span class="ne">AssertionError</span><span class="p">(</span><span class="s2">&quot;type stub not overridden&quot;</span><span class="p">)</span>
 
+    <span class="k">def</span> <span class="nf">__sym_ite__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">then_val</span><span class="p">,</span> <span class="n">else_val</span><span class="p">):</span>
+        <span class="k">raise</span> <span class="ne">AssertionError</span><span class="p">(</span><span class="s2">&quot;type stub not overridden&quot;</span><span class="p">)</span>
+
     <span class="k">def</span> <span class="fm">__eq__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">other</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">builtins</span><span class="o">.</span><span class="n">bool</span><span class="p">:</span>
         <span class="k">raise</span> <span class="ne">AssertionError</span><span class="p">(</span><span class="s2">&quot;type stub not overridden&quot;</span><span class="p">)</span>
 
@@ -928,6 +931,12 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
         <span class="k">return</span> <span class="n">b</span><span class="o">.</span><span class="n">__sym_min__</span><span class="p">(</span><span class="n">a</span><span class="p">)</span>
     <span class="k">return</span> <span class="n">builtins</span><span class="o">.</span><span class="n">min</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">)</span>  <span class="c1"># type: ignore[operator]</span></div>
 
+<div class="viewcode-block" id="sym_ite"><a class="viewcode-back" href="../generated/torch.sym_ite.html#torch.sym_ite">[docs]</a><span class="k">def</span> <span class="nf">sym_ite</span><span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="n">t</span><span class="p">,</span> <span class="n">f</span><span class="p">):</span>
+    <span class="k">assert</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="p">(</span><span class="n">SymBool</span><span class="p">,</span> <span class="n">builtins</span><span class="o">.</span><span class="n">bool</span><span class="p">))</span> <span class="ow">and</span> <span class="nb">type</span><span class="p">(</span><span class="n">t</span><span class="p">)</span> <span class="o">==</span> <span class="nb">type</span><span class="p">(</span><span class="n">f</span><span class="p">)</span>
+    <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="n">SymBool</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">b</span><span class="o">.</span><span class="n">__sym_ite__</span><span class="p">(</span><span class="n">t</span><span class="p">,</span> <span class="n">f</span><span class="p">)</span>
+    <span class="k">return</span> <span class="n">t</span> <span class="k">if</span> <span class="n">b</span> <span class="k">else</span> <span class="n">f</span></div>
+
 <span class="c1"># Check to see if we can load C extensions, and if not provide some guidance</span>
 <span class="c1"># on what the problem might be.</span>
 <span class="k">try</span><span class="p">:</span>

docs/main/_modules/torch.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/__config__.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_jit_internal.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_lobpcg.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 
@@ -486,6 +486,7 @@ <h1>Source code for torch._logging._internal</h1><div class="highlight"><pre>
 
 <span class="n">DEFAULT_LOG_LEVEL</span> <span class="o">=</span> <span class="n">logging</span><span class="o">.</span><span class="n">WARNING</span>
 <span class="n">LOG_ENV_VAR</span> <span class="o">=</span> <span class="s2">&quot;TORCH_LOGS&quot;</span>
+<span class="n">LOG_FORMAT_ENV_VAR</span> <span class="o">=</span> <span class="s2">&quot;TORCH_LOGS_FORMAT&quot;</span>
 
 
 <span class="nd">@dataclass</span>
@@ -640,6 +641,7 @@ <h1>Source code for torch._logging._internal</h1><div class="highlight"><pre>
     <span class="n">output_code</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
     <span class="n">schedule</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
     <span class="n">perf_hints</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
+    <span class="n">post_grad_graphs</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
     <span class="n">onnx_diagnostics</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
     <span class="n">fusion</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
     <span class="n">modules</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="n">Union</span><span class="p">[</span><span class="nb">int</span><span class="p">,</span> <span class="nb">bool</span><span class="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
@@ -755,6 +757,9 @@ <h1>Source code for torch._logging._internal</h1><div class="highlight"><pre>
 <span class="sd">        perf_hints (:class:`bool`):</span>
 <span class="sd">            Whether to emit the TorchInductor perf hints. Default: ``False``</span>
 
+<span class="sd">        post_grad_graphs (:class:`bool`):</span>
+<span class="sd">            Whether to emit the graphs generated by after post grad passes. Default: ``False``</span>
+
 <span class="sd">        onnx_diagnostics (:class:`bool`):</span>
 <span class="sd">            Whether to emit the ONNX exporter diagnostics in logging. Default: ``False``</span>
 
@@ -866,6 +871,7 @@ <h1>Source code for torch._logging._internal</h1><div class="highlight"><pre>
         <span class="n">output_code</span><span class="o">=</span><span class="n">output_code</span><span class="p">,</span>
         <span class="n">schedule</span><span class="o">=</span><span class="n">schedule</span><span class="p">,</span>
         <span class="n">perf_hints</span><span class="o">=</span><span class="n">perf_hints</span><span class="p">,</span>
+        <span class="n">post_grad_graphs</span><span class="o">=</span><span class="n">post_grad_graphs</span><span class="p">,</span>
         <span class="n">onnx</span><span class="o">=</span><span class="n">onnx</span><span class="p">,</span>
         <span class="n">onnx_diagnostics</span><span class="o">=</span><span class="n">onnx_diagnostics</span><span class="p">,</span>
         <span class="n">fusion</span><span class="o">=</span><span class="n">fusion</span><span class="p">,</span>
@@ -989,6 +995,11 @@ <h1>Source code for torch._logging._internal</h1><div class="highlight"><pre>
 
 <span class="s2">  TORCH_LOGS=&quot;+some.random.module,schedule&quot; will set the log level of</span>
 <span class="s2">  some.random.module to logging.DEBUG and enable the schedule artifact</span>
+
+<span class="s2">  TORCH_LOGS_FORMAT=&quot;</span><span class="si">%(levelname)s</span><span class="s2">: </span><span class="si">%(message)s</span><span class="s2">&quot; or any provided format</span>
+<span class="s2">  string will set the output format</span>
+<span class="s2">  Valid keys are &quot;levelname&quot;, &quot;message&quot;, &quot;pathname&quot;, &quot;levelno&quot;, &quot;lineno&quot;,</span>
+<span class="s2">  &quot;filename&quot; and &quot;name&quot;.</span>
 <span class="s2">&quot;&quot;&quot;</span>  <span class="c1"># flake8: noqa: B950</span>
     <span class="n">msg</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">&quot;&quot;&quot;</span>
 <span class="s2">TORCH_LOGS Info</span>
@@ -1137,7 +1148,15 @@ <h1>Source code for torch._logging._internal</h1><div class="highlight"><pre>
         <span class="k">return</span> <span class="s2">&quot;</span><span class="se">\n</span><span class="s2">&quot;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;</span><span class="si">{</span><span class="n">prefix</span><span class="si">}</span><span class="s2"> </span><span class="si">{</span><span class="n">l</span><span class="si">}</span><span class="s2">&quot;</span> <span class="k">for</span> <span class="n">l</span> <span class="ow">in</span> <span class="n">lines</span><span class="p">)</span>
 
 
-<span class="n">DEFAULT_FORMATTER</span> <span class="o">=</span> <span class="n">TorchLogsFormatter</span><span class="p">()</span>
+<span class="k">def</span> <span class="nf">_default_formatter</span><span class="p">():</span>
+    <span class="n">fmt</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">environ</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">LOG_FORMAT_ENV_VAR</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span>
+    <span class="k">if</span> <span class="n">fmt</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
+        <span class="k">return</span> <span class="n">TorchLogsFormatter</span><span class="p">()</span>
+    <span class="k">else</span><span class="p">:</span>
+        <span class="k">return</span> <span class="n">logging</span><span class="o">.</span><span class="n">Formatter</span><span class="p">(</span><span class="n">fmt</span><span class="p">)</span>
+
+
+<span class="n">DEFAULT_FORMATTER</span> <span class="o">=</span> <span class="n">_default_formatter</span><span class="p">()</span>
 
 
 <span class="k">def</span> <span class="nf">_setup_handlers</span><span class="p">(</span><span class="n">create_handler_fn</span><span class="p">,</span> <span class="n">log</span><span class="p">):</span>

docs/main/_modules/torch/_logging/_internal.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_lowrank.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_tensor.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_tensor_str.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_utils.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/amp/autocast_mode.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/modules/fused.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/qat/modules/conv_fused.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/qat/modules/linear_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/quantized/modules/bn_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/quantized/modules/conv_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/quantized/modules/linear_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/qat/dynamic/modules/linear.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/qat/modules/conv.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/qat/modules/linear.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantizable/modules/activation.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantizable/modules/rnn.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git2b2b6ca ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+git0a26e5f ) &#x25BC</a>
     </div>