Lines Matching refs:attempt
40 LOG(WARNING) << "Attempt to set current_global_step_ to smaller value: "
49 * Logs an attempt to extract the average gradient, and tries to flush all
51 * A TakeGrad attempt is blocked until num_required > counter_, i.e.,
55 * attempt is unblocked.
57 * callback: A callback to be executed after the attempt has been completed.
77 [this](Attempt* attempt) EXCLUSIVE_LOCKS_REQUIRED(mu_) {
78 if (counter_ >= attempt->elements_requested) {
80 attempt->context, attempt->done_callback);
111 for (Attempt& attempt : takegrad_attempts_) {
112 if (attempt.cancellation_manager == cancellation_manager &&
113 attempt.cancellation_token == token) {
114 if (!attempt.is_cancelled) {
115 attempt.is_cancelled = true;
116 attempt.context->SetStatus(
118 std::swap(callback, attempt.done_callback);
139 VLOG(1) << "Skipping cancelled TakeGrad attempt";
142 Attempt* cur_attempt = &takegrad_attempts_.front();
