quant_conv

Quantized convolution.

Classes

Conv2d

alias of QuantConv2d

QuantConv2d

Quantized 2D convolution.

Conv3d

alias of QuantConv3d

QuantConv3d

Quantized 3D convolution.

Conv1d

alias of QuantConv1d

QuantConv1d

Quantized 1D convolution.

ConvTranspose1d

alias of QuantConvTranspose1d

ConvTranspose2d

alias of QuantConvTranspose2d

ConvTranspose3d

alias of QuantConvTranspose3d

QuantConvTranspose1d

Quantized 1D transposed convolution.

QuantConvTranspose2d

Quantized 2D transposed convolution.

QuantConvTranspose3d

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')