quant_pooling — Model Optimizer 0.31.0 (original) (raw)

TensorRT Model Optimizer

Quantized Pooling modules.

Classes

AdaptiveAvgPool1d

alias of QuantAdaptiveAvgPool1d

AdaptiveAvgPool2d

alias of QuantAdaptiveAvgPool2d

AdaptiveAvgPool3d

alias of QuantAdaptiveAvgPool3d

AvgPool1d

alias of QuantAvgPool1d

AvgPool2d

alias of QuantAvgPool2d

AvgPool3d

alias of QuantAvgPool3d

MaxPool1d

alias of QuantMaxPool1d

MaxPool2d

alias of QuantMaxPool2d

MaxPool3d

alias of QuantMaxPool3d

class QuantAdaptiveAvgPool1d

Bases: _LegacyQuantInputBaseMixin, AdaptiveAvgPool1d

Quantized version of nn.AdaptiveAvgPool1d.

class QuantAdaptiveAvgPool2d

Bases: _LegacyQuantInputBaseMixin, AdaptiveAvgPool2d

Quantized version of nn.AdaptiveAvgPool2d.

class QuantAdaptiveAvgPool3d

Bases: _LegacyQuantInputBaseMixin, AdaptiveAvgPool3d

Quantized version of nn.AdaptiveAvgPool3d.

class QuantAvgPool1d

Bases: _LegacyQuantInputBaseMixin, AvgPool1d

Quantized version of nn.AvgPool1d.

class QuantAvgPool2d

Bases: _LegacyQuantInputBaseMixin, AvgPool2d

Quantized version of nn.AvgPool2d.

class QuantAvgPool3d

Bases: _LegacyQuantInputBaseMixin, AvgPool3d

Quantized version of nn.AvgPool3d.

class QuantMaxPool1d

Bases: _LegacyQuantInputBaseMixin, MaxPool1d

Quantized version of nn.MaxPool1d.

class QuantMaxPool2d

Bases: _LegacyQuantInputBaseMixin, MaxPool2d

Quantized version of nn.MaxPool2d.

class QuantMaxPool3d

Bases: _LegacyQuantInputBaseMixin, MaxPool3d

Quantized version of nn.MaxPool3d.