modules

Modules

modelopt.torch.quantization.nn.modules.clip

Implement a clip module as pytorch only has a simple clamp function.

modelopt.torch.quantization.nn.modules.quant_activations

Quantized activations module.

modelopt.torch.quantization.nn.modules.quant_batchnorm

Quantized batch normalization module.

modelopt.torch.quantization.nn.modules.quant_conv

Quantized convolution.

modelopt.torch.quantization.nn.modules.quant_instancenorm

Quantized instance normalization module.

modelopt.torch.quantization.nn.modules.quant_linear

Quantized Linear.

modelopt.torch.quantization.nn.modules.quant_module

Base class for quantization modules.

modelopt.torch.quantization.nn.modules.quant_pooling

Quantized Pooling modules.

modelopt.torch.quantization.nn.modules.quant_rnn

Quantized RNN.

modelopt.torch.quantization.nn.modules.tensor_quantizer

TensorQuantizer Module.

Modules with quantization support.