Home
last modified time | relevance | path

Searched refs:end_of_input (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/data/service/
Dtest_util_test.cc38 bool end_of_input = false; in TEST() local
41 while (!end_of_input) { in TEST()
43 TF_ASSERT_OK(iterator->GetNext(&outputs, &end_of_input)); in TEST()
44 if (!end_of_input) { in TEST()
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op.cc290 : end_of_input(false), in BatchResult()
297 bool end_of_input TF_GUARDED_BY(mu);
336 bool end_of_input = false; in CallBatching() local
337 for (int i = 0; i < dataset()->batch_size_ && !end_of_input; ++i) { in CallBatching()
340 &end_of_input); in CallBatching()
343 result->end_of_input = result->end_of_input || end_of_input; in CallBatching()
345 if (result->end_of_input || !result->status.ok()) break; in CallBatching()
347 if (!end_of_input) { in CallBatching()
358 DCHECK(end_of_input); in CallBatching()
445 result->end_of_input = reader->Contains( in ReadBatchResult()
[all …]
Dparallel_map_dataset_op.cc305 if (result.end_of_input) { in SaveInternal()
347 result.end_of_input = reader->Contains(element_prefix, kEndOfInput); in RestoreInternal()
380 bool end_of_input = false; member
428 &result->end_of_input); in CallFunction()
429 if (result->end_of_input || !result->status.ok()) { in CallFunction()
478 if (!result->end_of_input && result->status.ok()) { in ProcessResult()
499 *end_of_sequence = result->end_of_input; in ProcessResult()
558 (it == invocation_results_.begin() || !(*it)->end_of_input)) { in ShouldWait()
Dwindow_dataset_op.cc207 bool end_of_input; in GetNextInternal() local
210 input_impl_->GetNext(ctx, &element, &end_of_input).IgnoreError(); in GetNextInternal()
211 if (end_of_input) { in GetNextInternal()
Diterator_ops.cc730 bool end_of_input; in DoCompute() local
732 iterator->GetNext(&iter_ctx, &next_input_element, &end_of_input)); in DoCompute()
733 if (end_of_input) { in DoCompute()
Dparallel_interleave_dataset_op.cc951 bool end_of_input = false; in ProcessElement() local
953 &end_of_input); in ProcessElement()
954 if (end_of_input) { in ProcessElement()
/external/tensorflow/tensorflow/core/data/
Dstandalone_test.cc289 bool end_of_input = false; in TEST() local
290 for (int num_outputs = 0; !end_of_input; ++num_outputs) { in TEST()
292 s = iterator->GetNext(&outputs, &end_of_input); in TEST()
294 if (!end_of_input) { in TEST()
Dstandalone.cc38 Status Iterator::GetNext(std::vector<Tensor>* outputs, bool* end_of_input) { in GetNext() argument
39 return iterator_->GetNext(ctx_.get(), outputs, end_of_input); in GetNext()
Dstandalone.h73 Status GetNext(std::vector<Tensor>* outputs, bool* end_of_input);
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc340 : end_of_input(false), in BatchResult()
365 bool end_of_input TF_GUARDED_BY(mu);
402 bool end_of_input = false; in CallFunction() local
404 input_impl_->GetNext(ctx.get(), &input_element, &end_of_input); in CallFunction()
408 result->end_of_input = result->end_of_input || end_of_input; in CallFunction()
410 return_early = result->end_of_input || !result->status.ok(); in CallFunction()
597 result->end_of_input = reader->Contains( in ReadBatchResult()
622 if (result->end_of_input) { in WriteBatchResult()
Dsliding_window_dataset_op.cc193 bool end_of_input; in GetNextInternal() local
196 input_impl_->GetNext(ctx, &element, &end_of_input)); in GetNextInternal()
197 if (end_of_input) { in GetNextInternal()
Dparse_example_dataset_op.cc456 if (result.end_of_input) { in SaveInternal()
502 result.end_of_input = reader->Contains(full_name(strings::StrCat( in RestoreInternal()
536 bool end_of_input = false; member
585 &result->end_of_input); in CallFunction()
586 if (result->end_of_input || !result->status.ok()) { in CallFunction()
721 if (!result->end_of_input && result->status.ok()) { in ProcessResult()
735 *end_of_sequence = result->end_of_input; in ProcessResult()
797 (it == invocation_results_.begin() || !(*it)->end_of_input)) { in ShouldWait()
Dio_ops.cc103 bool end_of_input; in WriteData() local
104 TF_RETURN_IF_ERROR(iterator->GetNext(&iter_ctx, &element, &end_of_input)); in WriteData()
105 if (end_of_input) { in WriteData()
Dparallel_interleave_dataset_op.cc357 bool end_of_input = false; in GetNextInternal() local
358 Status s = input_impl_->GetNext(ctx, &args, &end_of_input); in GetNextInternal()
359 if (end_of_input) { in GetNextInternal()
663 bool end_of_input = false; in EnsureWorkerThreadsStarted() local
664 Status s = input_impl_->GetNext(ctx, &args, &end_of_input); in EnsureWorkerThreadsStarted()
665 if (end_of_input) { in EnsureWorkerThreadsStarted()
/external/webrtc/modules/audio_coding/neteq/tools/
Dneteq_quality_test.cc442 bool end_of_input = false; in Simulate() local
447 while (!end_of_input && decoded_time_ms_ < runtime_ms) { in Simulate()
453 end_of_input = true; in Simulate()
454 ASSERT_TRUE(end_of_input && absl::GetFlag(FLAGS_runtime_ms) < 0); in Simulate()
/external/tensorflow/tensorflow/core/kernels/
Dlookup_util.cc424 bool end_of_input; in Next() local
426 status_ = iterator_->GetNext(iterator_ctx_.get(), &tensors_, &end_of_input); in Next()
427 if (status_.ok() && end_of_input) { in Next()
/external/libwebm/m2ts/
Dvpxpes_parser.cc309 const std::uint8_t* const end_of_input = in RemoveStartCodeEmulationPreventionBytes() local
313 if (raw_data + i > end_of_input) in RemoveStartCodeEmulationPreventionBytes()