megengine.module.quant_dequant 源代码

from .module import Module


[文档]class QuantStub(Module): r"""A helper :class:`~.Module` simply returning input. Could be replaced with :class:`~.QATModule` version :class:`~.qat.QuantStub` using :func:`~.quantize.quantize_qat`. """ def forward(self, inp): return inp
[文档]class DequantStub(Module): r"""A helper :class:`~.Module` simply returning input. Could be replaced with :class:`~.QATModule` version :class:`~.qat.DequantStub` using :func:`~.quantize.quantize_qat`. """ def forward(self, inp): return inp