modules — Model Optimizer 0.31.0 (original) (raw)
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.