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_embedding
|
Quantized Embedding. |
modelopt.torch.quantization.nn.modules.quant_instancenorm
|
Quantized instance normalization module. |
modelopt.torch.quantization.nn.modules.quant_layernorm
|
Registers torch.nn.LayerNorm with QuantInputBase. |
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. |