Lines Matching refs:attempt
83 [tuple, this](Attempt* attempt) EXCLUSIVE_LOCKS_REQUIRED(mu_) {
85 attempt->context->SetStatus(
91 attempt->context->SetStatus(errors::InvalidArgument(
148 [tuple, this, ctx](Attempt* attempt) EXCLUSIVE_LOCKS_REQUIRED(mu_) {
150 attempt->context->SetStatus(
158 tuple[0].dim_size(0) - attempt->elements_requested;
161 attempt->context->SetStatus(GetElementComponentFromBatch(
162 tuple, index, 0, attempt->context, &priority_element));
163 if (!attempt->context->status().ok()) return kComplete;
166 attempt->context->SetStatus(errors::InvalidArgument(
175 attempt->context->SetStatus(GetElementComponentFromBatch(
176 tuple, index, i, attempt->context, &element));
177 if (!attempt->context->status().ok()) return kComplete;
180 --attempt->elements_requested;
181 if (attempt->elements_requested == 0) {
210 [callback, this](Attempt* attempt) EXCLUSIVE_LOCKS_REQUIRED(mu_) {
213 attempt->context->SetStatus(errors::OutOfRange(
221 DequeueLocked(attempt->context, &tuple);
222 attempt->done_callback = [callback, tuple]() { callback(tuple); };
302 allow_small_batch](Attempt* attempt) EXCLUSIVE_LOCKS_REQUIRED(mu_) {
310 (!allow_small_batch && s < attempt->elements_requested)) {
311 attempt->context->SetStatus(errors::OutOfRange(
314 attempt->elements_requested, ", current size ", s, ")"));
325 // An alternative solution is to store the attempt tuple
330 if (s < attempt->elements_requested) {
341 if (attempt->tuple.empty()) {
345 attempt->tuple.reserve(num_components());
348 ManyOutShape(i, attempt->elements_requested);
350 attempt->context->SetStatus(attempt->context->allocate_temp(
352 if (!attempt->context->status().ok()) return kComplete;
353 attempt->tuple.emplace_back(element);
358 DequeueLocked(attempt->context, &tuple);
360 attempt->tuple[0].dim_size(0) - attempt->elements_requested;
362 attempt->context->SetStatus(batch_util::CopyElementToSlice(
363 std::move(tuple[i]), &attempt->tuple[i], index));
364 if (!attempt->context->status().ok()) return kComplete;
367 --attempt->elements_requested;
368 if (attempt->elements_requested == 0) {
369 tuple = attempt->tuple;
370 attempt->done_callback = [callback, tuple]() {
