quark.torch.quantization.nn.modules

quark.torch.quantization.nn.modules#

Package Contents#

Classes#

QuantConv2d

Quantized version of nn.Conv2d

QuantConvTranspose2d

Quantized version of nn.ConvTranspose2d

QuantLinear

Quantized version of nn.Linear

QuantEmbedding

QuantEmbeddingBag