QuantStub

class torch.ao.quantization.QuantStub(qconfig=None) [source]

Quantize stub module, before calibration, this is same as an observer, it will be swapped as nnq.Quantize in convert.

Parameters

qconfig (Optional[QConfig]) – quantization configuration for the tensor, if qconfig is not provided, we will get qconfig from parent modules