Skip to content

Latest commit

 

History

History
29 lines (21 loc) · 473 Bytes

torch.nn.intrinsic.qat.rst

File metadata and controls

29 lines (21 loc) · 473 Bytes

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:

LinearReLU

.. autoclass:: LinearReLU
    :members: