def forward(self, x): output = inplace_abn(x, self.weight, self.bias, self.running_mean, self.running_var, self.training, self.momentum, self.eps, self.act_name, self.act_param) if isinstance(output, tuple): output = output[0] return output
def inplace_abn_sync(**kwargs): inplace_abn(**kwargs)