quark.torch.export.nn.modules.export_operator

Module Contents

Classes

class quark.torch.export.nn.modules.export_operator.ExportLinear(quant_linear: quark.torch.quantization.nn.modules.quantize_linear.QuantLinear, reorder: bool = True, custom_mode: Optional[str] = None)

Exporting version of nn.Linear

class quark.torch.export.nn.modules.export_operator.ExportConv2d(quant_conv2d: quark.torch.quantization.nn.modules.quantize_conv.QuantConv2d, reorder: bool = True, custom_mode: Optional[str] = None)

Exporting version of nn.Conv2d