def _center_bias_fn():
    delta_updates = array_ops.where((bias_hess > 0), ((- bias_grads) / bias_hess), array_ops.zeros_like(bias_grads))
    center_bias = training_ops.center_tree_ensemble_bias(tree_ensemble_handle=self._ensemble_handle, stamp_token=ensemble_stamp, next_stamp_token=next_ensemble_stamp, delta_updates=delta_updates, learner_config=self._learner_config_serialized)
    return training_state.continue_centering.assign(center_bias)
def _center_bias_fn():
    delta_updates = array_ops.where((bias_hess > 0), ((- bias_grads) / bias_hess), array_ops.zeros_like(bias_grads))
    center_bias = training_ops.center_tree_ensemble_bias(tree_ensemble_handle=self._ensemble_handle, stamp_token=ensemble_stamp, next_stamp_token=next_ensemble_stamp, delta_updates=delta_updates, learner_config=self._learner_config_serialized)
    return training_state.continue_centering.assign(center_bias)