Rate this Page

ConvAdd2d#

class torch.ao.nn.intrinsic.modules.fused.ConvAdd2d(conv, add)[source]#

This is a sequential container which calls the Conv2d modules with extra Add. During quantization this will be replaced with the corresponding fused module.

forward(x1, x2)[source]#

Applies convolution to x1 and adds the result to x2.