quant_conv
Quantized convolution.
Classes
alias of |
|
Quantized 2D convolution. |
|
alias of |
|
Quantized 3D convolution. |
|
alias of |
|
Quantized 1D convolution. |
|
alias of |
|
alias of |
|
alias of |
|
Quantized 1D transposed convolution. |
|
Quantized 2D transposed convolution. |
|
Quantized 3D transposed convolution. |
- Conv1d
alias of
QuantConv1d
- Conv2d
alias of
QuantConv2d
- Conv3d
alias of
QuantConv3d
- ConvTranspose1d
alias of
QuantConvTranspose1d
- ConvTranspose2d
alias of
QuantConvTranspose2d
- ConvTranspose3d
alias of
QuantConvTranspose3d
- class QuantConv1d
Bases:
_LegacyQuantLinearConvBaseMixin
,Conv1d
Quantized 1D convolution.
- default_quant_desc_weight = QuantizerAttributeConfig(enable=True, num_bits=8, axis=0, fake_quant=True, unsigned=False, narrow_range=False, learn_amax=False, type='static', block_sizes=None, trt_high_precision_dtype='Float', calibrator='max')
- class QuantConv2d
Bases:
_LegacyQuantLinearConvBaseMixin
,Conv2d
Quantized 2D convolution.
- default_quant_desc_weight = QuantizerAttributeConfig(enable=True, num_bits=8, axis=0, fake_quant=True, unsigned=False, narrow_range=False, learn_amax=False, type='static', block_sizes=None, trt_high_precision_dtype='Float', calibrator='max')
- class QuantConv3d
Bases:
_LegacyQuantLinearConvBaseMixin
,Conv3d
Quantized 3D convolution.
- default_quant_desc_weight = QuantizerAttributeConfig(enable=True, num_bits=8, axis=0, fake_quant=True, unsigned=False, narrow_range=False, learn_amax=False, type='static', block_sizes=None, trt_high_precision_dtype='Float', calibrator='max')
- class QuantConvTranspose1d
Bases:
_LegacyQuantLinearConvBaseMixin
,ConvTranspose1d
Quantized 1D transposed convolution.
- default_quant_desc_weight = QuantizerAttributeConfig(enable=True, num_bits=8, axis=0, fake_quant=True, unsigned=False, narrow_range=False, learn_amax=False, type='static', block_sizes=None, trt_high_precision_dtype='Float', calibrator='max')
- class QuantConvTranspose2d
Bases:
_LegacyQuantLinearConvBaseMixin
,ConvTranspose2d
Quantized 2D transposed convolution.
- default_quant_desc_weight = QuantizerAttributeConfig(enable=True, num_bits=8, axis=0, fake_quant=True, unsigned=False, narrow_range=False, learn_amax=False, type='static', block_sizes=None, trt_high_precision_dtype='Float', calibrator='max')
- class QuantConvTranspose3d
Bases:
_LegacyQuantLinearConvBaseMixin
,ConvTranspose3d
Quantized 3D transposed convolution.
- default_quant_desc_weight = QuantizerAttributeConfig(enable=True, num_bits=8, axis=0, fake_quant=True, unsigned=False, narrow_range=False, learn_amax=False, type='static', block_sizes=None, trt_high_precision_dtype='Float', calibrator='max')