megengine.module.quantized.batch_matmul_activation.
BatchMatMulActivation
基类:megengine.module.batch_matmul_activation.BatchMatMulActivation, megengine.module.quantized.module.QuantizedModule
megengine.module.batch_matmul_activation.BatchMatMulActivation
megengine.module.quantized.module.QuantizedModule
calc_bmm_quantized
forward
from_qat_module
返回从 QATModule 实例转换而来的 QuantizedModule 实例。
QATModule
QuantizedModule
megengine.module.quantized.concat.
Concat
基类:megengine.module.quantized.module.QuantizedModule
量化(quantized)版本concat的 QuantizedModule,仅用于推理阶段。
megengine.module.quantized.conv.
Conv2d
基类:megengine.module.conv.Conv2d, megengine.module.quantized.module.QuantizedModule
megengine.module.conv.Conv2d
量化(quantized)版本 Conv2d。
calc_conv_quantized
ConvRelu2d
基类:megengine.module.quantized.conv.Conv2d
megengine.module.quantized.conv.Conv2d
量化(quantized)版本 ConvRelu2d。
megengine.module.quantized.conv_bn.
ConvBn2d
基类:megengine.module.quantized.conv_bn._ConvBnActivation2d
megengine.module.quantized.conv_bn._ConvBnActivation2d
量化(quantized)版本 ConvBn2d。
ConvBnRelu2d
量化(quantized)版本 ConvBnRelu2d。
megengine.module.quantized.elemwise.
Elemwise
量化(quantized)版本 Elemwise。
megengine.module.quantized.linear.
Linear
量化(quantized)版本 Linear。
megengine.module.quantized.module.
基类:megengine.module.module.Module
megengine.module.module.Module
量化(quantized)版本Module的基类。应从QATModule转换而来,不支持训练。
megengine.module.quantized.quant_dequant.
DequantStub
量化(quantized)版本 DequantStub,可将量化后的输入重置为float32类型。
QuantStub
量化(quantized)版本 QuantStub,可将输入转化为量化数据类型。