| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | threadpool_dataset_op.cc | 205 Status Initialize(IteratorContext* ctx) override { in Initialize()
207 IteratorContext(CreateParams(ctx)), this, prefix(), &input_impl_); in Initialize()
210 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
213 return input_impl_->GetNext(IteratorContext(CreateParams(ctx)), in GetNextInternal()
219 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
231 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
238 IteratorContext::Params CreateParams(IteratorContext* ctx) { in CreateParams()
240 IteratorContext::Params params(ctx); in CreateParams()
337 Status Initialize(IteratorContext* ctx) override { in Initialize()
342 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
[all …]
|
| D | set_stats_aggregator_dataset_op.cc | 175 Status Initialize(IteratorContext* ctx) override { in Initialize()
176 IteratorContext iter_ctx = ContextWithAggregator(ctx); in Initialize()
181 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
185 IteratorContext iter_ctx = ContextWithAggregator(ctx); in GetNextInternal()
189 IteratorContext ContextWithAggregator(IteratorContext* ctx) { in ContextWithAggregator()
192 IteratorContext::Params params(ctx); in ContextWithAggregator()
197 IteratorContext iter_ctx(std::move(params)); in ContextWithAggregator()
203 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
214 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | stats_dataset_ops.cc | 109 Status Initialize(IteratorContext* ctx) override { in Initialize()
114 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
132 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
144 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
226 Status Initialize(IteratorContext* ctx) override { in Initialize()
231 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
251 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
263 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | snapshot_dataset_op.cc | 171 Status Initialize(IteratorContext* ctx) override;
173 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
180 Status RestoreInternal(IteratorContext* ctx,
184 Status InitializeIterator(IteratorContext* ctx, IteratorStateReader* reader);
207 Status Initialize(IteratorContext* ctx) override;
209 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
216 Status RestoreInternal(IteratorContext* ctx,
244 Status Initialize(IteratorContext* ctx) override;
246 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
253 Status RestoreInternal(IteratorContext* ctx,
[all …]
|
| D | non_serializable_dataset_op.cc | 98 Status Initialize(IteratorContext* ctx) override { in Initialize()
103 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
111 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
121 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | ignore_errors_dataset_op.cc | 99 Status Initialize(IteratorContext* ctx) override { in Initialize()
104 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
133 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
149 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | sleep_dataset_op.cc | 116 Status Initialize(IteratorContext* ctx) override { in Initialize()
128 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
146 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
156 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | io_ops.cc | 76 IteratorContext::Params params(ctx); in WriteData()
85 IteratorContext iter_ctx(std::move(params)); in WriteData()
141 Status SaveDatasetOp::GetShardIndex(IteratorContext* ctx, in GetShardIndex()
258 Status Initialize(IteratorContext* ctx) override { in Initialize()
266 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
275 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
284 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
290 Status InitializeInput(IteratorContext* ctx) in InitializeInput()
|
| D | choose_fastest_branch_dataset_op.cc | 89 Status Initialize(IteratorContext* ctx) override { in Initialize()
99 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
108 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
117 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
344 Status Initialize(IteratorContext* ctx) override { in Initialize()
360 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
399 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
428 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
456 Status GetNextFromExperiment(IteratorContext* ctx, in GetNextFromExperiment()
498 Status MakeCurrentIterator(IteratorContext* ctx, int64 branch_index, in MakeCurrentIterator()
|
| D | assert_cardinality_dataset_op.cc | 99 Status Initialize(IteratorContext* ctx) override { in Initialize()
103 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
129 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
142 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | assert_next_dataset_op.cc | 95 Status Initialize(IteratorContext* ctx) override { in Initialize()
117 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
125 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
136 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | map_and_batch_dataset_op.cc | 216 Status Initialize(IteratorContext* ctx) override { in Initialize()
230 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
271 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
299 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
376 void CallCompleted(const std::shared_ptr<IteratorContext>& ctx, in CallCompleted()
393 void CallFunction(std::shared_ptr<IteratorContext> ctx, in CallFunction()
487 void EnsureRunnerThreadStarted(IteratorContext* ctx) in EnsureRunnerThreadStarted()
490 auto ctx_copy = std::make_shared<IteratorContext>(*ctx); in EnsureRunnerThreadStarted()
498 const std::shared_ptr<IteratorContext>& ctx, in EnsureOutputAllocated()
525 void RunnerThread(const std::shared_ptr<IteratorContext>& ctx) in RunnerThread()
[all …]
|
| D | take_while_dataset_op.cc | 128 Status Initialize(IteratorContext* ctx) override { in Initialize()
135 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
170 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
188 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.h | 338 class IteratorContext {
341 explicit Params(IteratorContext* ctx) in Params()
432 explicit IteratorContext(IteratorContext* ctx) : params_(Params{ctx}) {} in IteratorContext() function
434 explicit IteratorContext(OpKernelContext* ctx) : params_(Params{ctx}) {} in IteratorContext() function
436 explicit IteratorContext(Params params) : params_(std::move(params)) {} in IteratorContext() function
627 virtual Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
630 Status GetNext(IteratorContext&& ctx, std::vector<Tensor>* out_tensors, in GetNext()
642 virtual Status Skip(IteratorContext* ctx, int num_to_skip,
661 virtual Status Initialize(IteratorContext* ctx) { return Status::OK(); } in Initialize()
664 Status InitializeBase(IteratorContext* ctx, const IteratorBase* parent);
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.h | 46 IteratorContext* ctx, const IteratorBase* parent,
55 IteratorContext* ctx, const IteratorBase* parent,
155 Status Instantiate(IteratorContext* ctx,
192 Status IsMultiDevice(IteratorContext* ctx, bool* is_multi_device) const;
225 Status Run(IteratorContext* ctx, std::vector<Tensor>&& args,
233 Status Run(IteratorContext* ctx, std::vector<Tensor>&& args,
240 Status RunWithBorrowedArgs(IteratorContext* ctx,
248 Status RunWithBorrowedArgs(IteratorContext* ctx,
267 void RunAsync(IteratorContext* ctx, std::vector<Tensor>&& args,
|
| D | repeat_dataset_op.cc | 118 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
127 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
136 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
147 Status Initialize(IteratorContext* ctx) override { in Initialize()
151 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
179 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
196 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
221 Status Initialize(IteratorContext* ctx) override { in Initialize()
226 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
260 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
[all …]
|
| D | model_dataset_op.cc | 134 Status Initialize(IteratorContext* ctx) override { in Initialize()
135 IteratorContext::Params params(ctx); in Initialize()
137 return dataset()->input_->MakeIterator(IteratorContext(std::move(params)), in Initialize()
141 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
144 IteratorContext::Params params(ctx); in GetNextInternal()
152 Status s = input_impl_->GetNext(IteratorContext(std::move(params)), in GetNextInternal()
162 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
174 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
176 IteratorContext::Params params(ctx); in RestoreInternal()
179 TF_RETURN_IF_ERROR(RestoreInput(IteratorContext(std::move(params)), in RestoreInternal()
[all …]
|
| D | skip_dataset_op.cc | 104 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
113 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
123 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
134 Status Initialize(IteratorContext* ctx) override { in Initialize()
138 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
171 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
188 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | take_dataset_op.cc | 91 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNextInternal()
99 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
109 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
120 Status Initialize(IteratorContext* ctx) override { in Initialize()
124 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNextInternal()
147 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
164 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | cache_dataset_ops.cc | 151 Status Initialize(IteratorContext* ctx) override { in Initialize()
156 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
165 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
176 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
251 Status Initialize(IteratorContext* ctx) override { in Initialize()
256 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
305 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
346 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
497 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
535 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
[all …]
|
| D | parallel_batch_dataset_op.cc | 181 Status Initialize(IteratorContext* ctx) override { in Initialize()
192 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
233 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
258 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
305 void CallCompleted(const std::shared_ptr<IteratorContext>& ctx, in CallCompleted()
317 void CallBatching(std::shared_ptr<IteratorContext> ctx, in CallBatching()
388 void EnsureRunnerThreadStarted(IteratorContext* ctx) in EnsureRunnerThreadStarted()
391 auto ctx_copy = std::make_shared<IteratorContext>(*ctx); in EnsureRunnerThreadStarted()
398 void RunnerThread(const std::shared_ptr<IteratorContext>& ctx) in RunnerThread()
439 Status ReadBatchResult(IteratorContext* ctx, IteratorStateReader* reader, in ReadBatchResult()
|
| D | parallel_map_dataset_op.cc | 221 Status Initialize(IteratorContext* ctx) override { in Initialize()
228 IteratorContext::Params params(ctx); in Initialize()
234 IteratorContext(params), this, prefix(), &input_impl_)); in Initialize()
239 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
267 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
313 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
394 void EnsureThreadsStarted(IteratorContext* ctx) in EnsureThreadsStarted()
397 auto ctx_copy = std::make_shared<IteratorContext>(*ctx); in EnsureThreadsStarted()
409 void CallCompleted(const std::shared_ptr<IteratorContext>& ctx, in CallCompleted()
418 void CallFunction(const std::shared_ptr<IteratorContext>& ctx, in CallFunction()
[all …]
|
| D | flat_map_dataset_op.cc | 118 Status Initialize(IteratorContext* ctx) override { in Initialize()
125 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
167 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
200 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
244 Status BuildCurrentElementIteratorLocked(IteratorContext* ctx, in BuildCurrentElementIteratorLocked()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 42 Iterator::Iterator(IteratorBase* iterator, IteratorContext* ctx) in Iterator()
105 std::unique_ptr<IteratorContext> ctx; in MakeIterator()
115 IteratorContext::Params params(&op_ctx); in MakeIterator()
121 ctx = absl::make_unique<IteratorContext>(std::move(params)); in MakeIterator()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/sql/ |
| D | query_connection.h | 23 class IteratorContext; variable
64 virtual Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
|