¶

megengine.module.quantized 模块

megengine.module.quantized.batch_matmul_activation

class megengine.module.quantized.batch_matmul_activation.BatchMatMulActivation(batch, in_features, out_features, bias=True, nonlinear_mode='IDENTITY', dtype=None, **kwargs)

基类：megengine.module.batch_matmul_activation.BatchMatMulActivation, megengine.module.quantized.module.QuantizedModule

calc_bmm_quantized(inp)

forward(inp)

classmethod from_qat_module(qat_module)

返回从 QATModule 实例转换而来的 QuantizedModule 实例。

megengine.module.quantized.concat

class megengine.module.quantized.concat.Concat(dtype=None)

基类：megengine.module.quantized.module.QuantizedModule

量化（quantized）版本concat的 QuantizedModule，仅用于推理阶段。

forward(inps, axis=0)

classmethod from_qat_module(qat_module)

返回从 QATModule 实例转换而来的 QuantizedModule 实例。

megengine.module.quantized.conv

class megengine.module.quantized.conv.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, conv_mode='CROSS_CORRELATION', compute_mode='DEFAULT', dtype=None)

基类：megengine.module.conv.Conv2d, megengine.module.quantized.module.QuantizedModule

量化（quantized）版本 Conv2d。

calc_conv_quantized(inp, nonlinear_mode='IDENTITY')

forward(inp)

classmethod from_qat_module(qat_module)

返回从 QATModule 实例转换而来的 QuantizedModule 实例。

class megengine.module.quantized.conv.ConvRelu2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, conv_mode='CROSS_CORRELATION', compute_mode='DEFAULT', dtype=None)

基类：megengine.module.quantized.conv.Conv2d

量化（quantized）版本 ConvRelu2d。

forward(inp)

megengine.module.quantized.conv_bn

class megengine.module.quantized.conv_bn.ConvBn2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, conv_mode='CROSS_CORRELATION', compute_mode='DEFAULT', dtype=None)

基类：megengine.module.quantized.conv_bn._ConvBnActivation2d

量化（quantized）版本 ConvBn2d。

forward(inp)

class megengine.module.quantized.conv_bn.ConvBnRelu2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, conv_mode='CROSS_CORRELATION', compute_mode='DEFAULT', dtype=None)

基类：megengine.module.quantized.conv_bn._ConvBnActivation2d

量化（quantized）版本 ConvBnRelu2d。

forward(inp)

megengine.module.quantized.elemwise

class megengine.module.quantized.elemwise.Elemwise(method, dtype=None)

基类：megengine.module.quantized.module.QuantizedModule

量化（quantized）版本 Elemwise。

forward(*inps)

classmethod from_qat_module(qat_module)

返回从 QATModule 实例转换而来的 QuantizedModule 实例。

megengine.module.quantized.linear

class megengine.module.quantized.linear.Linear(dtype=None)

基类：megengine.module.quantized.module.QuantizedModule

量化（quantized）版本 Linear。

forward(inp)

classmethod from_qat_module(qat_module)

返回从 QATModule 实例转换而来的 QuantizedModule 实例。

megengine.module.quantized.module

class megengine.module.quantized.module.QuantizedModule

基类：megengine.module.module.Module

量化（quantized）版本Module的基类。应从QATModule转换而来，不支持训练。

abstract classmethod from_qat_module(qat_module)

返回从 QATModule 实例转换而来的 QuantizedModule 实例。

megengine.module.quantized.quant_dequant

class megengine.module.quantized.quant_dequant.DequantStub

基类：megengine.module.quantized.module.QuantizedModule

量化（quantized）版本 DequantStub，可将量化后的输入重置为float32类型。

forward(inp)

classmethod from_qat_module(qat_module)

返回从 QATModule 实例转换而来的 QuantizedModule 实例。

class megengine.module.quantized.quant_dequant.QuantStub(dtype=None)

基类：megengine.module.quantized.module.QuantizedModule

量化（quantized）版本 QuantStub，可将输入转化为量化数据类型。

forward(inp)

classmethod from_qat_module(qat_module)

返回从 QATModule 实例转换而来的 QuantizedModule 实例。