ConvRelu2d

class ConvRelu2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, conv_mode='cross_correlation', compute_mode='default', padding_mode='zeros', **kwargs)

A fused Module including Conv2d and relu. Could be replaced with QATModule version ConvRelu2d using quantize_qat.

previous

PixelShuffle

next

ConvBn2d