r"""A helper :class:`~.Module` simply returning input. Could be replaced with :class:`~.QATModule` version :class:`~.qat.QuantStub` using :func:`~.quantize.quantize_qat`.
| 2 | |
| 3 | |
| 4 | class QuantStub(Module): |
| 5 | r"""A helper :class:`~.Module` simply returning input. Could be replaced with :class:`~.QATModule` |
| 6 | version :class:`~.qat.QuantStub` using :func:`~.quantize.quantize_qat`. |
| 7 | """ |
| 8 | |
| 9 | def forward(self, inp): |
| 10 | return inp |
| 11 | |
| 12 | |
| 13 | class DequantStub(Module): |