Skip to content

Latest commit

 

History

History
43 lines (33 loc) · 676 Bytes

torch.nn.intrinsic.qat.rst

File metadata and controls

43 lines (33 loc) · 676 Bytes
Raw

torch.nn.intrinsic.qat

This module implements the versions of those fused operations needed for quantization aware training.

.. automodule:: torch.nn.intrinsic.qat

ConvBn2d

.. autoclass:: ConvBn2d
    :members:

ConvBnReLU2d

.. autoclass:: ConvBnReLU2d
    :members:

ConvReLU2d

.. autoclass:: ConvReLU2d
    :members:

ConvBn3d

.. autoclass:: ConvBn3d
    :members:

ConvBnReLU3d

.. autoclass:: ConvBnReLU3d
    :members:

ConvReLU3d

.. autoclass:: ConvReLU3d
    :members:

LinearReLU

.. autoclass:: LinearReLU
    :members: