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: