| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | cache_dataset_ops_test.cc | 171 bool end_of_sequence = false; in TEST_P() local
173 while (!end_of_sequence) { in TEST_P()
176 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
186 end_of_sequence = false; in TEST_P()
188 while (!end_of_sequence) { in TEST_P()
191 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
312 bool end_of_sequence = false; in TEST_P() local
317 while (!end_of_sequence) { in TEST_P()
319 &end_of_sequence)); in TEST_P()
321 end_of_sequence = false; in TEST_P()
[all …]
|
| D | repeat_dataset_op_test.cc | 138 bool end_of_sequence = false; in TEST_P() local
146 &end_of_sequence)); in TEST_P()
158 EXPECT_FALSE(end_of_sequence); in TEST_P()
160 while (!end_of_sequence) { in TEST_P()
162 &end_of_sequence)); in TEST_P()
163 if (!end_of_sequence) { in TEST_P()
316 bool end_of_sequence = dataset_->Cardinality() == 0; in TEST_P() local
333 &end_of_sequence)); in TEST_P()
334 if (!end_of_sequence) { in TEST_P()
350 EXPECT_FALSE(end_of_sequence); in TEST_P()
[all …]
|
| D | repeat_dataset_op.cc | 115 bool* end_of_sequence) override { in GetNextInternal() argument
116 *end_of_sequence = true; in GetNextInternal()
147 bool* end_of_sequence) override { in GetNextInternal() argument
150 *end_of_sequence = true; in GetNextInternal()
155 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
156 if (!*end_of_sequence) { in GetNextInternal()
163 *end_of_sequence = true; in GetNextInternal()
218 bool* end_of_sequence) override { in GetNextInternal() argument
225 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
226 DCHECK(!*end_of_sequence || out_tensors->empty()); in GetNextInternal()
[all …]
|
| D | skip_dataset_op.cc | 101 bool* end_of_sequence) override { in GetNextInternal() argument
102 *end_of_sequence = true; in GetNextInternal()
134 bool* end_of_sequence) override { in GetNextInternal() argument
138 *end_of_sequence = true; in GetNextInternal()
149 input_impl_->GetNext(ctx, &dummy_out_tensors, end_of_sequence)); in GetNextInternal()
150 if (*end_of_sequence) { in GetNextInternal()
161 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
162 if (*end_of_sequence) { in GetNextInternal()
|
| D | batch_dataset_op.cc | 145 bool* end_of_sequence) override { in GetNextInternal() argument
152 *end_of_sequence = true; in GetNextInternal()
156 *end_of_sequence = false; in GetNextInternal()
157 for (int i = 0; i < dataset()->batch_size_ && !*end_of_sequence; ++i) { in GetNextInternal()
160 input_impl_->GetNext(ctx, &batch_element_tuple, end_of_sequence)); in GetNextInternal()
161 if (!*end_of_sequence) { in GetNextInternal()
170 DCHECK(*end_of_sequence); in GetNextInternal()
176 *end_of_sequence = true; in GetNextInternal()
249 *end_of_sequence = false; in GetNextInternal()
|
| D | take_dataset_op.cc | 86 bool* end_of_sequence) override { in GetNextInternal() argument
87 *end_of_sequence = true; in GetNextInternal()
118 bool* end_of_sequence) override { in GetNextInternal() argument
121 *end_of_sequence = true; in GetNextInternal()
126 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
127 if (!*end_of_sequence) { in GetNextInternal()
133 *end_of_sequence = true; in GetNextInternal()
|
| D | zip_dataset_op.cc | 131 bool* end_of_sequence) override { in GetNextInternal() argument
134 *end_of_sequence = true; in GetNextInternal()
140 *end_of_sequence = false; in GetNextInternal()
146 *end_of_sequence |= component_end_of_sequence; in GetNextInternal()
152 if (*end_of_sequence) { in GetNextInternal()
158 if (*end_of_sequence || !status.ok()) { in GetNextInternal()
161 if (*end_of_sequence) { in GetNextInternal()
|
| D | cache_dataset_ops.cc | 154 bool* end_of_sequence) override { in GetNextInternal() argument
156 return iterator_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
252 bool* end_of_sequence) override { in GetNextInternal() argument
254 *end_of_sequence = false; in GetNextInternal()
255 TF_RETURN_IF_ERROR(EnsureLockFileExists(end_of_sequence)); in GetNextInternal()
256 if (*end_of_sequence) { in GetNextInternal()
272 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
273 if (*end_of_sequence && out_tensors->empty()) { in GetNextInternal()
290 if (*end_of_sequence) { in GetNextInternal()
376 Status EnsureLockFileExists(bool* end_of_sequence) in EnsureLockFileExists() argument
[all …]
|
| D | sparse_tensor_slice_dataset_op_test.cc | 168 bool end_of_sequence = false; in TEST_P() local
171 while (!end_of_sequence) { in TEST_P()
173 &end_of_sequence)); in TEST_P()
174 if (!end_of_sequence) { in TEST_P()
375 bool end_of_sequence = false; in TEST_P() local
382 &end_of_sequence)); in TEST_P()
387 EXPECT_FALSE(end_of_sequence); in TEST_P()
398 EXPECT_TRUE(end_of_sequence); in TEST_P()
|
| D | tensor_dataset_op_test.cc | 126 bool end_of_sequence = false; in TEST_P() local
128 while (!end_of_sequence) { in TEST_P()
130 &end_of_sequence)); in TEST_P()
232 bool end_of_sequence = false; in TEST_P() local
250 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
256 EXPECT_TRUE(end_of_sequence); in TEST_P()
258 EXPECT_FALSE(end_of_sequence); in TEST_P()
|
| D | shard_dataset_op.cc | 123 bool* end_of_sequence) override { in GetNextInternal() argument
127 *end_of_sequence = true; in GetNextInternal()
134 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &result, end_of_sequence)); in GetNextInternal()
135 if (*end_of_sequence) { in GetNextInternal()
145 Status s = input_impl_->GetNext(ctx, &unused_result, end_of_sequence); in GetNextInternal()
146 if (*end_of_sequence || errors::IsOutOfRange(s)) { in GetNextInternal()
|
| D | padded_batch_dataset_op.cc | 190 bool* end_of_sequence) override { in GetNextInternal() argument
197 *end_of_sequence = true; in GetNextInternal()
200 *end_of_sequence = false; in GetNextInternal()
202 for (int i = 0; i < dataset()->batch_size_ && !*end_of_sequence; in GetNextInternal()
206 end_of_sequence)); in GetNextInternal()
207 if (!*end_of_sequence) { in GetNextInternal()
211 if (*end_of_sequence) { in GetNextInternal()
218 DCHECK(*end_of_sequence); in GetNextInternal()
224 *end_of_sequence = true; in GetNextInternal()
337 *end_of_sequence = false; in GetNextInternal()
|
| D | tensor_slice_dataset_op_test.cc | 106 bool end_of_sequence = false; in TEST_P() local
110 while (!end_of_sequence) { in TEST_P()
112 &end_of_sequence)); in TEST_P()
268 bool end_of_sequence = false; in TEST_P() local
279 &end_of_sequence)); in TEST_P()
284 EXPECT_FALSE(end_of_sequence); in TEST_P()
305 EXPECT_TRUE(end_of_sequence); in TEST_P()
|
| D | shuffle_dataset_op_test.cc | 327 bool end_of_sequence = false; in TEST_P() local
329 while (!end_of_sequence) { in TEST_P()
332 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
345 end_of_sequence = false; in TEST_P()
350 while (!end_of_sequence) { in TEST_P()
353 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
523 bool end_of_sequence = false; in TEST_P() local
540 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
|
| D | window_dataset_op_test.cc | 289 bool end_of_sequence = false; in TEST_P() local
291 while (!end_of_sequence) { in TEST_P()
296 &end_of_sequence)); in TEST_P()
297 if (!end_of_sequence) { in TEST_P()
471 bool end_of_sequence = false; in TEST_P() local
484 while (!end_of_sequence) { in TEST_P()
489 &end_of_sequence)); in TEST_P()
490 if (!end_of_sequence) { in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unbatch_dataset_op.cc | 101 bool* end_of_sequence) override { in GetNextInternal() argument
104 *end_of_sequence = true; in GetNextInternal()
107 *end_of_sequence = false; in GetNextInternal()
108 while (!*end_of_sequence) { in GetNextInternal()
119 *end_of_sequence = false; in GetNextInternal()
126 input_impl_->GetNext(ctx, &tensors_, end_of_sequence)); in GetNextInternal()
127 if (!*end_of_sequence) { in GetNextInternal()
|
| D | take_while_dataset_op.cc | 133 bool* end_of_sequence) override { in GetNextInternal() argument
137 *end_of_sequence = true; in GetNextInternal()
141 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
143 if (*end_of_sequence) { in GetNextInternal()
157 *end_of_sequence = !result[0].scalar<bool>()(); in GetNextInternal()
158 if (*end_of_sequence) { in GetNextInternal()
|
| D | ignore_errors_dataset_op.cc | 89 bool* end_of_sequence) override { in GetNextInternal() argument
94 *end_of_sequence = true; in GetNextInternal()
97 s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
100 s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
103 if (*end_of_sequence) { in GetNextInternal()
|
| D | sliding_window_dataset_op.cc | 145 bool* end_of_sequence) override { in GetNextInternal() argument
153 *end_of_sequence = true; in GetNextInternal()
160 *end_of_sequence = false; in GetNextInternal()
161 for (size_t i = buffer_.size(); i < target_size && !*end_of_sequence; in GetNextInternal()
165 input_impl_->GetNext(ctx, &element, end_of_sequence)); in GetNextInternal()
166 if (!*end_of_sequence) { in GetNextInternal()
175 DCHECK(*end_of_sequence); in GetNextInternal()
230 *end_of_sequence = false; in GetNextInternal()
|
| D | to_tf_record_op.cc | 89 bool end_of_sequence; in DoCompute() local
92 iterator->GetNext(&iter_ctx, &components, &end_of_sequence)); in DoCompute()
94 if (!end_of_sequence) { in DoCompute()
99 } while (!end_of_sequence); in DoCompute()
|
| D | dense_to_sparse_batch_dataset_op.cc | 160 bool* end_of_sequence) override { in GetNextInternal() argument
185 *end_of_sequence = false; in GetNextInternal()
188 !*end_of_sequence; in GetNextInternal()
192 end_of_sequence)); in GetNextInternal()
193 if (!*end_of_sequence) { in GetNextInternal()
227 DCHECK(*end_of_sequence); in GetNextInternal()
280 *end_of_sequence = false; in GetNextInternal()
|
| D | stats_dataset_ops.cc | 109 bool* end_of_sequence) override { in GetNextInternal() argument
112 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
115 if (stats_aggregator && !*end_of_sequence) { in GetNextInternal()
224 bool* end_of_sequence) override { in GetNextInternal() argument
226 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
228 if (stats_aggregator && s.ok() && !*end_of_sequence) { in GetNextInternal()
|
| D | parallel_interleave_dataset_op.cc | 251 bool* end_of_sequence) override { in GetNextInternal() argument
284 *end_of_sequence = false; in GetNextInternal()
333 *end_of_sequence = true; in GetNextInternal()
575 bool end_of_sequence = false; member
760 bool end_of_sequence = false; in WorkerThread() local
761 while (!end_of_sequence) { in WorkerThread()
768 !worker_thread_states_[thread_index].end_of_sequence) { in WorkerThread()
773 &worker_thread_states_[thread_index].end_of_sequence); in WorkerThread()
774 end_of_sequence = in WorkerThread()
775 worker_thread_states_[thread_index].end_of_sequence; in WorkerThread()
[all …]
|
| D | directed_interleave_dataset_op.cc | 158 bool* end_of_sequence) override { in GetNextInternal() argument
161 *end_of_sequence = true; in GetNextInternal()
167 *end_of_sequence = false; in GetNextInternal()
169 ctx, &selector_result, end_of_sequence)); in GetNextInternal()
170 if (*end_of_sequence) { in GetNextInternal()
200 *end_of_sequence = true; in GetNextInternal()
|
| D | sampling_dataset_op.cc | 110 bool* end_of_sequence) override { in GetNextInternal() argument
116 *end_of_sequence = true; in GetNextInternal()
120 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
122 if (*end_of_sequence) { in GetNextInternal()
136 *end_of_sequence = false; in GetNextInternal()
|