in inplace_abn/abn.py [0:0]
def forward(self, x: torch.Tensor) -> torch.Tensor:
momentum, training = self._get_momentum_and_training()
running_mean, running_var = self._get_running_stats()
return inplace_abn(
x,
self.weight,
self.bias,
running_mean,
running_var,
training,
momentum,
self.eps,
self.activation,
self.activation_param,
)