MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / ConvTransposeRelu2d

Class ConvTransposeRelu2d

imperative/python/megengine/module/conv.py:821–827  ·  view source on GitHub ↗

r"""A fused :class:`~.Module` including :class:`~.module.ConvTranspose2d` and :func:`~.relu`. Could be replaced with :class:`~.QATModule` version :class:`~.qat.ConvTransposeRelu2d` using :func:`~.quantize.quantize_qat`.

Source from the content-addressed store, hash-verified

819
820
821class ConvTransposeRelu2d(ConvTranspose2d):
822 r"""A fused :class:`~.Module` including :class:`~.module.ConvTranspose2d` and :func:`~.relu`.
823 Could be replaced with :class:`~.QATModule` version :class:`~.qat.ConvTransposeRelu2d` using :func:`~.quantize.quantize_qat`.
824 """
825
826 def forward(self, inp):
827 return relu(self.calc_conv_transpose2d(inp, self.weight, self.bias))
828
829
830class DeformableConv2d(_ConvNd):

Callers 1

__init__Method · 0.90

Calls

no outgoing calls

Tested by 1

__init__Method · 0.72