def METHOD_NAME(self, batch_mean, batch_var, batch_count):
    """Updates from batch mean, variance and count moments."""
    self.mean, self.var, self.count = update_mean_var_count_from_moments(
        self.mean, self.var, self.count, batch_mean, batch_var, batch_count
    )