QuantStub

class QuantStub(**kwargs)[source]

A helper QATModule simply return input, but will quantize input after converted to QuantizedModule.

classmethod from_float_module(float_module)[source]

Return a QATModule instance converted from a float Module instance.

previous

BatchMatMulActivation

next

DequantStub