megengine.module.quantized package

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 实例。