mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
59cb693c90
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/51770 Test Plan: tested locally on mac Imported from OSS Reviewed By: jerryzh168 Differential Revision: D26279112 fbshipit-source-id: 8675d3ef712ecbe545bad0d3502181b3ccdd7f89
125 lines
1.9 KiB
ReStructuredText
125 lines
1.9 KiB
ReStructuredText
torch.nn.quantized
|
|
------------------
|
|
|
|
This module implements the quantized versions of the nn modules and functionals.
|
|
|
|
Functional interface
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: torch.nn.quantized.functional
|
|
|
|
.. autofunction:: linear
|
|
.. autofunction:: conv1d
|
|
.. autofunction:: conv2d
|
|
.. autofunction:: conv3d
|
|
.. autofunction:: max_pool2d
|
|
.. autofunction:: adaptive_avg_pool2d
|
|
.. autofunction:: avg_pool2d
|
|
.. autofunction:: interpolate
|
|
.. autofunction:: hardswish
|
|
.. autofunction:: upsample
|
|
.. autofunction:: upsample_bilinear
|
|
.. autofunction:: upsample_nearest
|
|
|
|
|
|
.. automodule:: torch.nn.quantized
|
|
|
|
ReLU6
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: ReLU6
|
|
:members:
|
|
|
|
ELU
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: ELU
|
|
:members:
|
|
|
|
Hardswish
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: Hardswish
|
|
:members:
|
|
|
|
Conv1d
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: Conv1d
|
|
:members:
|
|
|
|
Conv2d
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: Conv2d
|
|
:members:
|
|
|
|
Conv3d
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: Conv3d
|
|
:members:
|
|
|
|
FloatFunctional
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: FloatFunctional
|
|
:members:
|
|
|
|
QFunctional
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: QFunctional
|
|
:members:
|
|
|
|
Quantize
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: Quantize
|
|
:members:
|
|
|
|
DeQuantize
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: DeQuantize
|
|
:members:
|
|
|
|
Linear
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: Linear
|
|
:members:
|
|
|
|
BatchNorm2d
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: BatchNorm2d
|
|
:members:
|
|
|
|
BatchNorm3d
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: BatchNorm3d
|
|
:members:
|
|
|
|
LayerNorm
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: LayerNorm
|
|
:members:
|
|
|
|
GroupNorm
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: GroupNorm
|
|
:members:
|
|
|
|
InstanceNorm1d
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: InstanceNorm1d
|
|
:members:
|
|
|
|
InstanceNorm2d
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: InstanceNorm2d
|
|
:members:
|
|
|
|
InstanceNorm3d
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: InstanceNorm3d
|
|
:members:
|
|
|
|
Embedding
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: Embedding
|
|
:members:
|
|
|
|
EmbeddingBag
|
|
~~~~~~~~~~~~~~~
|
|
.. autoclass:: EmbeddingBag
|
|
:members:
|