quark.torch.export.nn.modules.export_operator

Contents

`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, 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, custom_mode: Optional[str] = None)#: Exporting version of nn.Conv2d