Skip to content

Latest commit

 

History

History
124 lines (99 loc) · 1.89 KB

torch.nn.quantized.rst

File metadata and controls

124 lines (99 loc) · 1.89 KB
Raw

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: