neural_compressor.torch.quantization.modules

Torch.nn.Module Class Definition.

Classes

FakeAffineTensorQuantFunction	Fake version of affine quantization.
TEQLinearFakeQuant	Wrapper quantization linear.
MulLinear	Linear wrapper to apply scale to input.

Module Contents

class neural_compressor.torch.quantization.modules.FakeAffineTensorQuantFunction

Fake version of affine quantization.

class neural_compressor.torch.quantization.modules.TEQLinearFakeQuant(orig_layer, alpha=None, num_bits=4, group_size=-1, scheme='asym')

Wrapper quantization linear.

class neural_compressor.torch.quantization.modules.MulLinear(module, input_scale=None)

Linear wrapper to apply scale to input.