Searched refs:end_of_input (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone_test.cc | 170 bool end_of_input = false; in TEST() local
171 for (int num_outputs = 0; !end_of_input; ++num_outputs) { in TEST()
173 s = iterator->GetNext(&outputs, &end_of_input); in TEST()
175 if (!end_of_input) { in TEST()
|
| D | standalone.cc | 35 Status Iterator::GetNext(std::vector<Tensor>* outputs, bool* end_of_input) { in GetNext() argument
36 return iterator_->GetNext(ctx_.get(), outputs, end_of_input); in GetNext()
|
| D | standalone.h | 73 Status GetNext(std::vector<Tensor>* outputs, bool* end_of_input);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_map_iterator.cc | 134 if (result.end_of_input) { in SaveInternal()
175 result.end_of_input = reader->Contains(full_name( in RestoreInternal()
187 bool end_of_input; member
223 input_impl_->GetNext(ctx.get(), &input_element, &result->end_of_input); in CallFunction()
224 if (result->end_of_input || !result->status.ok()) { in CallFunction()
245 if (!result->end_of_input && result->status.ok()) { in ProcessResult()
266 *end_of_sequence = result->end_of_input; in ProcessResult()
324 (it == invocation_results_.begin() || !(*it)->end_of_input)) { in ShouldWait()
|
| D | window_dataset_op.cc | 197 bool end_of_input; in GetNextInternal() local
200 input_impl_->GetNext(ctx, &element, &end_of_input).IgnoreError(); in GetNextInternal()
201 if (end_of_input) { in GetNextInternal()
|
| D | parallel_interleave_dataset_op.cc | 538 bool end_of_input = false; in FetchResults() local
542 ctx.get(), &result->return_values, &end_of_input); in FetchResults()
543 if (end_of_input) { in FetchResults()
557 if (end_of_input) { in FetchResults()
|
| D | iterator_ops.cc | 701 bool end_of_input; in ComputeAsync() local
703 &end_of_input); in ComputeAsync()
704 if (!status.ok() || end_of_input) { in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op.cc | 360 end_of_input = false; in BatchResult()
385 bool end_of_input GUARDED_BY(mu);
416 bool end_of_input; in CallFunction() local
418 input_impl_->GetNext(ctx.get(), &input_element, &end_of_input); in CallFunction()
422 result->end_of_input = result->end_of_input || end_of_input; in CallFunction()
424 return_early = result->end_of_input || !result->status.ok(); in CallFunction()
668 result->end_of_input = reader->Contains( in ReadBatchResult()
730 if (result->end_of_input) { in WriteBatchResult()
|
| D | sliding_window_dataset_op.cc | 184 bool end_of_input; in GetNextInternal() local
187 input_impl_->GetNext(ctx, &element, &end_of_input)); in GetNextInternal()
188 if (end_of_input) { in GetNextInternal()
|
| D | parallel_interleave_dataset_op.cc | 320 bool end_of_input = false; in GetNextInternal() local
321 Status s = input_impl_->GetNext(ctx, &args, &end_of_input); in GetNextInternal()
322 if (end_of_input) { in GetNextInternal()
587 bool end_of_input = false; in EnsureWorkerThreadsStarted() local
588 Status s = input_impl_->GetNext(ctx, &args, &end_of_input); in EnsureWorkerThreadsStarted()
589 if (end_of_input) { in EnsureWorkerThreadsStarted()
|