| 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: |
