We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 197e6b2 commit 99e65c9Copy full SHA for 99e65c9
prototype_source/fx_graph_mode_quant_guide.rst
@@ -30,6 +30,7 @@ Symbolically trace only the code that needs to be quantized
30
-----------------------------------------------------------------
31
When the whole model is not symbolically traceable but the submodule we want to quantize is
32
symbolically traceable, we can run quantization only on that submodule.
33
+
34
before:
35
36
.. code:: python
@@ -44,6 +45,7 @@ before:
44
45
after:
46
47
48
49
class FP32Traceable(nn.Module):
50
def forward(self, x):
51
x = traceable_code(x)
0 commit comments