| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | repeat_dataset_op.cc | 148 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
155 if (!input_impl_) { in GetNextInternal()
161 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
170 dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_)); in GetNextInternal()
173 input_impl_.reset(); in GetNextInternal()
188 if (!input_impl_) { in SaveInternal()
191 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
201 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
203 input_impl_.reset(); in RestoreInternal()
211 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
[all …]
|
| D | concatenate_dataset_op.cc | 121 ctx, this, strings::StrCat(prefix(), "[0]"), &input_impl_); in Initialize()
128 if (!input_impl_) { in GetNextInternal()
134 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
140 ctx, this, strings::StrCat(prefix(), "[1]"), &input_impl_)); in GetNextInternal()
144 input_impl_.reset(); in GetNextInternal()
159 if (input_impl_) { in SaveInternal()
160 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
173 input_impl_.reset(); in RestoreInternal()
180 ctx, this, strings::StrCat(prefix(), "[1]"), &input_impl_)); in RestoreInternal()
182 input_impl_.reset(); in RestoreInternal()
[all …]
|
| D | skip_dataset_op.cc | 135 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
143 if (!input_impl_) { in GetNextInternal()
150 TF_RETURN_IF_ERROR(input_impl_->Skip(ctx, dataset()->count_ - i_, in GetNextInternal()
155 input_impl_.reset(); in GetNextInternal()
162 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
164 input_impl_.reset(); in GetNextInternal()
180 if (input_impl_) { in SaveInternal()
181 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
193 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
195 input_impl_.reset(); in RestoreInternal()
[all …]
|
| D | shard_dataset_op.cc | 123 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
132 if (!input_impl_) { in GetNextInternal()
144 input_impl_->Skip(ctx, num_to_skip, end_of_sequence, &num_skipped)); in GetNextInternal()
147 input_impl_.reset(); in GetNextInternal()
152 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &result, end_of_sequence)); in GetNextInternal()
154 input_impl_.reset(); in GetNextInternal()
162 Status s = input_impl_->Skip(ctx, dataset()->num_shards_ - next_index_, in GetNextInternal()
199 if (!input_impl_) { in SaveInternal()
202 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
213 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
[all …]
|
| D | take_dataset_op.cc | 121 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
127 if (!input_impl_) { in GetNextInternal()
133 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
141 input_impl_.reset(); in GetNextInternal()
156 if (input_impl_) { in SaveInternal()
157 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
169 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
171 input_impl_.reset(); in RestoreInternal()
179 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | filter_dataset_op.cc | 119 dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_)); in Initialize()
136 if (!input_impl_) { in GetNextInternal()
141 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
145 input_impl_.reset(); in GetNextInternal()
207 if (input_impl_) in SaveInternal()
208 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
223 input_impl_.reset(); in RestoreInternal()
225 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
235 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | flat_map_dataset_op.cc | 120 dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_)); in Initialize()
130 if (!input_impl_) { in GetNextInternal()
154 input_impl_->GetNext(ctx, &captured_func_inputs_, end_of_sequence)); in GetNextInternal()
156 input_impl_.reset(); in GetNextInternal()
176 if (input_impl_) { in SaveInternal()
177 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
203 input_impl_.reset(); in RestoreInternal()
209 dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_)); in RestoreInternal()
210 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
263 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | batch_dataset_op.cc | 154 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
165 if (!input_impl_) { in GetNextInternal()
174 input_impl_->GetNext(ctx, &batch_element_tuple, end_of_sequence)); in GetNextInternal()
178 input_impl_.reset(); in GetNextInternal()
218 if (!input_impl_) { in SaveInternal()
221 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
230 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
232 input_impl_.reset(); in RestoreInternal()
243 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | window_dataset_op.cc | 146 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
161 if (!input_impl_ && in GetNextInternal()
169 if (input_impl_) { in GetNextInternal()
175 input_impl_->GetNext(ctx, &element, end_of_sequence); in GetNextInternal()
180 input_impl_.reset(); in GetNextInternal()
206 for (size_t i = buffer_size; input_impl_ && i < window_shift; ++i) { in GetNextInternal()
210 input_impl_->GetNext(ctx, &element, &end_of_input).IgnoreError(); in GetNextInternal()
212 input_impl_.reset(); in GetNextInternal()
269 if (!input_impl_) { in SaveInternal()
272 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
[all …]
|
| D | padded_batch_dataset_op.cc | 189 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
200 if (!input_impl_) { in GetNextInternal()
209 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &batch_element_tuple, in GetNextInternal()
216 input_impl_.reset(); in GetNextInternal()
354 if (input_impl_) in SaveInternal()
355 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
365 input_impl_.reset(); in RestoreInternal()
368 dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_)); in RestoreInternal()
369 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
380 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | map_dataset_op.cc | 142 dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_)); in Initialize()
156 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &args, end_of_sequence)); in GetNextInternal()
192 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
198 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
203 std::unique_ptr<IteratorBase> input_impl_; member in tensorflow::data::MapDatasetOp::Dataset::Iterator
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | ignore_errors_dataset_op.cc | 101 &input_impl_); in Initialize()
110 if (!input_impl_) { in GetNextInternal()
114 s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
121 s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
126 input_impl_.reset(); in GetNextInternal()
141 if (input_impl_) in SaveInternal()
142 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
153 input_impl_.reset(); in RestoreInternal()
155 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
161 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | take_while_dataset_op.cc | 130 dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_)); in Initialize()
140 if (!input_impl_) { in GetNextInternal()
145 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
149 input_impl_.reset(); in GetNextInternal()
180 if (input_impl_) in SaveInternal()
181 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
192 input_impl_.reset(); in RestoreInternal()
194 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
200 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | threadpool_dataset_op.cc | 207 IteratorContext(CreateParams(ctx)), this, prefix(), &input_impl_); in Initialize()
213 return input_impl_->GetNext(IteratorContext(CreateParams(ctx)), in GetNextInternal()
226 DCHECK(input_impl_ != nullptr); in SaveInternal()
227 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
233 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
248 std::unique_ptr<IteratorBase> input_impl_; member in tensorflow::data::experimental::__anon3c251a260111::ThreadPoolDatasetOp::Dataset::Iterator
339 &input_impl_); in Initialize()
349 return input_impl_->GetNext(IteratorContext{std::move(params)}, in GetNextInternal()
362 DCHECK(input_impl_ != nullptr); in SaveInternal()
363 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
[all …]
|
| D | sampling_dataset_op.cc | 109 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
119 if (!input_impl_) { in GetNextInternal()
124 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
128 input_impl_.reset(); in GetNextInternal()
164 if (input_impl_) { in SaveInternal()
165 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
187 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
189 input_impl_.reset(); in RestoreInternal()
198 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | sliding_window_dataset_op.cc | 147 &input_impl_); in Initialize()
159 if (!input_impl_) { in GetNextInternal()
172 input_impl_->GetNext(ctx, &element, end_of_sequence)); in GetNextInternal()
176 input_impl_.reset(); in GetNextInternal()
196 input_impl_->GetNext(ctx, &element, &end_of_input)); in GetNextInternal()
198 input_impl_.reset(); in GetNextInternal()
251 if (!input_impl_) { in SaveInternal()
255 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
275 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
277 input_impl_.reset(); in RestoreInternal()
[all …]
|
| D | stats_dataset_ops.cc | 111 &input_impl_); in Initialize()
119 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
140 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
147 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
153 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
228 &input_impl_); in Initialize()
235 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
259 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
266 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
272 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | unbatch_dataset_op.cc | 118 &input_impl_); in Initialize()
125 if (!input_impl_) { in GetNextInternal()
148 input_impl_->GetNext(ctx, &tensors_, end_of_sequence)); in GetNextInternal()
169 input_impl_.reset(); in GetNextInternal()
191 if (input_impl_) { in SaveInternal()
192 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
214 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
216 input_impl_.reset(); in RestoreInternal()
240 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | unique_dataset_op.cc | 83 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
95 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
114 if (input_impl_) { in SaveInternal()
115 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
134 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
136 input_impl_.reset(); in RestoreInternal()
205 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | rebatch_dataset_op.cc | 121 &input_impl_); in Initialize()
133 input_impl_->GetNext(ctx, &input_tensors, end_of_sequence)); in GetNextInternal()
186 if (!input_impl_) { in SaveInternal()
190 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
210 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
212 input_impl_.reset(); in RestoreInternal()
254 std::unique_ptr<IteratorBase> input_impl_; member in tensorflow::data::experimental::__anon714a254d0111::RebatchDatasetOp::Dataset::Iterator
378 &input_impl_); in Initialize()
405 input_impl_->GetNext(ctx, &tensors_, &end_of_sequence_)); in GetNextInternal()
524 if (!input_impl_) { in SaveInternal()
[all …]
|
| D | non_serializable_dataset_op.cc | 100 &input_impl_); in Initialize()
106 return input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
117 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
123 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
128 std::unique_ptr<IteratorBase> input_impl_; member in tensorflow::data::experimental::__anon5926aa950111::NonSerializableDatasetOp::Dataset::Iterator
|
| D | sleep_dataset_op.cc | 125 &input_impl_); in Initialize()
141 return input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
153 return SaveInput(ctx, writer, input_impl_); in SaveInternal()
158 return RestoreInput(ctx, reader, input_impl_); in RestoreInternal()
162 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|
| D | assert_cardinality_dataset_op.cc | 100 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
107 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
138 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
146 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
158 std::unique_ptr<IteratorBase> input_impl_; member in tensorflow::data::experimental::AssertCardinalityDatasetOp::Dataset::Iterator
|
| D | assert_next_dataset_op.cc | 114 return dataset()->input_->MakeIterator(ctx, this, prefix(), &input_impl_); in Initialize()
120 return input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
132 TF_RETURN_IF_ERROR(SaveInput(ctx, writer, input_impl_)); in SaveInternal()
138 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
143 std::unique_ptr<IteratorBase> input_impl_; member in tensorflow::data::experimental::AssertNextDatasetOp::Dataset::Iterator
|
| D | set_stats_aggregator_dataset_op.cc | 178 &input_impl_); in Initialize()
186 return input_impl_->GetNext(&iter_ctx, out_tensors, end_of_sequence); in GetNextInternal()
211 return SaveInput(ctx, writer, input_impl_); in SaveInternal()
217 return RestoreInput(ctx, reader, input_impl_); in RestoreInternal()
222 std::unique_ptr<IteratorBase> input_impl_ TF_GUARDED_BY(mu_);
|