torch.nn.quantized#

FloatFunctional

State collector class for float operations.

FXFloatFunctional

module to replace FloatFunctional module before FX graph mode quantization, since activation_post_process will be inserted in top level module directly

QFunctional

Wrapper class for quantized operations.