modules

Modules

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.