in quant/models/resnet.py [0:0]
def forward(self, x: torch.Tensor) -> torch.Tensor: # type: ignore
"""Forward pass of XnorBasicBlock."""
out1 = self.nonlin1(self.conv1(self.bn1(x)))
if self.double_shortcut:
out1 = out1 + self.shortcut(x)
out2 = self.conv2(self.bn2(out1))
if self.double_shortcut:
out2 = self.nonlin2(out2)
return out2 + out1
out2 = out2 + self.shortcut(x)
return self.nonlin2(out2)