torch.nn.qat --------------------------- This module implements versions of the key nn modules **Conv2d()** and **Linear()** which run in FP32 but with rounding applied to simulate the effect of INT8 quantization. .. automodule:: torch.nn.qat Conv2d ~~~~~~~~~~~~~~~ .. autoclass:: Conv2d :members: Linear ~~~~~~~~~~~~~~~ .. autoclass:: Linear :members: