r"""A fused :class:`~.Module` including :class:`~.module.Conv2d` and :func:`~.relu`. Could be replaced with :class:`~.QATModule` version :class:`~.qat.ConvRelu2d` using :func:`~.quantize.quantize_qat`.
| 810 | |
| 811 | |
| 812 | class ConvRelu2d(Conv2d): |
| 813 | r"""A fused :class:`~.Module` including :class:`~.module.Conv2d` and :func:`~.relu`. |
| 814 | Could be replaced with :class:`~.QATModule` version :class:`~.qat.ConvRelu2d` using :func:`~.quantize.quantize_qat`. |
| 815 | """ |
| 816 | |
| 817 | def forward(self, inp): |
| 818 | return relu(self.calc_conv(inp, self.weight, self.bias)) |
| 819 | |
| 820 | |
| 821 | class ConvTransposeRelu2d(ConvTranspose2d): |