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