docs/source/torch.nn.quantized.rst

torch.nn.quantized
------------------

This module implements the quantized versions of the nn layers such as
~`torch.nn.Conv2d` and `torch.nn.ReLU`.

Functional interface
~~~~~~~~~~~~~~~~~~~~
.. automodule:: torch.nn.quantized.functional

.. autofunction:: relu
.. 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

ReLU
~~~~~~~~~~~~~~~
.. autoclass:: ReLU
    :members:

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: