| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | model_dataset_op.cc | 107 Status Initialize(IteratorContext* ctx) override { in Initialize()
108 IteratorContext::Params params(ctx); in Initialize()
111 IteratorContext(std::move(params)), prefix(), &input_impl_); in Initialize()
114 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
117 IteratorContext::Params params(ctx); in GetNextInternal()
123 return input_impl_->GetNext(IteratorContext(std::move(params)), in GetNextInternal()
129 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
140 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
148 Status EnsureOptimizeThreadStarted(IteratorContext* ctx) in EnsureOptimizeThreadStarted()
151 std::shared_ptr<IteratorContext> new_ctx = in EnsureOptimizeThreadStarted()
[all …]
|
| D | repeat_dataset_op.cc | 109 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
118 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
126 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
137 Status Initialize(IteratorContext* ctx) override { in Initialize()
141 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
166 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
183 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
208 Status Initialize(IteratorContext* ctx) override { in Initialize()
213 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
243 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
[all …]
|
| D | take_dataset_op.cc | 46 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNextInternal()
54 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
63 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
74 Status Initialize(IteratorContext* ctx) override { in Initialize()
78 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNextInternal()
101 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
118 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | cache_dataset_ops.cc | 126 Status Initialize(IteratorContext* ctx) override { in Initialize()
131 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
140 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
150 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
207 Status Initialize(IteratorContext* ctx) override { in Initialize()
211 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
256 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
296 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
447 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
488 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
[all …]
|
| D | skip_dataset_op.cc | 96 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
105 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
114 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
125 Status Initialize(IteratorContext* ctx) override { in Initialize()
129 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
167 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
184 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | graph_rewrite_dataset.cc | 201 Status Initialize(IteratorContext* ctx) override { in Initialize()
202 IteratorContext::Params params(ctx); in Initialize()
206 IteratorContext(std::move(params)), prefix(), &input_impl_); in Initialize()
209 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNextInternal()
211 IteratorContext::Params params(ctx); in GetNextInternal()
214 return input_impl_->GetNext(IteratorContext(std::move(params)), out_tensors, in GetNextInternal()
220 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
230 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | parallel_map_iterator.cc | 75 Status Initialize(IteratorContext* ctx) override { in Initialize()
85 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNextInternal()
105 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
144 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
190 void EnsureRunnerThreadStarted(IteratorContext* ctx) in EnsureRunnerThreadStarted()
193 auto ctx_copy = std::make_shared<IteratorContext>(*ctx); in EnsureRunnerThreadStarted()
200 void CallCompleted(const std::shared_ptr<IteratorContext>& ctx, in CallCompleted()
217 void CallFunction(const std::shared_ptr<IteratorContext>& ctx, in CallFunction()
241 Status ProcessResult(IteratorContext* ctx, in ProcessResult()
270 void RunnerThread(const std::shared_ptr<IteratorContext>& ctx) in RunnerThread()
|
| D | prefetch_dataset_op.cc | 101 Status Initialize(IteratorContext* ctx) override { in Initialize()
105 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
154 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
185 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
231 Status Consume(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in Consume()
266 Status EnsurePrefetchThreadStarted(IteratorContext* ctx) in EnsurePrefetchThreadStarted()
269 std::shared_ptr<IteratorContext> new_ctx = in EnsurePrefetchThreadStarted()
270 std::make_shared<IteratorContext>(*ctx); in EnsurePrefetchThreadStarted()
281 void PrefetchThread(const std::shared_ptr<IteratorContext>& ctx) { in PrefetchThread()
|
| D | filter_dataset_op.cc | 37 std::function<Status(IteratorContext*, InstantiatedCapturedFunction*,
58 filter_pred = [](IteratorContext* ctx, in MakeDataset()
76 IteratorContext* ctx, in MakeDataset()
173 Status Initialize(IteratorContext* ctx) override { in Initialize()
180 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
247 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
265 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | captured_function.h | 57 Status Run(IteratorContext* ctx, std::vector<Tensor>&& args,
63 Status RunWithBorrowedArgs(IteratorContext* ctx,
79 void RunAsync(IteratorContext* ctx, std::vector<Tensor>&& args,
144 Status Instantiate(IteratorContext* ctx,
|
| D | map_dataset_op.cc | 33 std::function<Status(IteratorContext*, InstantiatedCapturedFunction*,
57 map_func = [](IteratorContext* ctx, in MakeDataset()
67 IteratorContext* ctx, in MakeDataset()
197 Status Initialize(IteratorContext* ctx) override { in Initialize()
204 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
241 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
251 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | filter_by_component_dataset_op.cc | 106 Status Initialize(IteratorContext* ctx) override { in Initialize()
110 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 | shard_dataset_op.cc | 116 Status Initialize(IteratorContext* ctx) override { in Initialize()
120 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
147 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
165 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | threadpool_dataset_op.cc | 194 Status Initialize(IteratorContext* ctx) override { in Initialize()
196 IteratorContext(CreateParams(ctx)), prefix(), &input_impl_); in Initialize()
199 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
202 return input_impl_->GetNext(IteratorContext(CreateParams(ctx)), in GetNextInternal()
208 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
214 IteratorContext::Params CreateParams(IteratorContext* ctx) { in CreateParams()
216 IteratorContext::Params params(ctx); in CreateParams()
302 Status Initialize(IteratorContext* ctx) override { in Initialize()
306 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
309 IteratorContext::Params params(ctx); in GetNextInternal()
[all …]
|
| D | stats_dataset_ops.cc | 98 Status Initialize(IteratorContext* ctx) override { in Initialize()
102 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
119 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
130 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
208 Status Initialize(IteratorContext* ctx) override { in Initialize()
212 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
231 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
242 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | take_while_dataset_op.cc | 36 std::function<Status(IteratorContext*, InstantiatedCapturedFunction*,
57 loop_pred = [](IteratorContext* ctx, in MakeDataset()
75 IteratorContext* ctx, in MakeDataset()
174 Status Initialize(IteratorContext* ctx) override { in Initialize()
181 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
204 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
219 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | set_stats_aggregator_dataset_op.cc | 159 Status Initialize(IteratorContext* ctx) override { in Initialize()
163 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
169 IteratorContext::Params params(ctx); in GetNextInternal()
174 IteratorContext iter_ctx(std::move(params)); in GetNextInternal()
180 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
190 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | non_serializable_dataset_op.cc | 86 Status Initialize(IteratorContext* ctx) override { in Initialize()
90 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
98 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
107 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | sleep_dataset_op.cc | 97 Status Initialize(IteratorContext* ctx) override { in Initialize()
101 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
112 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
121 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | choose_fastest_branch_dataset_op.cc | 81 Status Initialize(IteratorContext* ctx) override { in Initialize()
91 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
100 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
108 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
326 Status Initialize(IteratorContext* ctx) override { in Initialize()
342 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
379 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
407 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
433 Status GetNextFromExperiment(IteratorContext* ctx, in GetNextFromExperiment()
471 Status MakeCurrentIterator(IteratorContext* ctx, int64 branch_index, in MakeCurrentIterator()
|
| D | ignore_errors_dataset_op.cc | 81 Status Initialize(IteratorContext* ctx) override { in Initialize()
85 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
109 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
124 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| D | map_and_batch_dataset_op.cc | 52 std::function<void(IteratorContext*, InstantiatedCapturedFunction*,
95 map_func = [](IteratorContext* ctx, in MakeDataset()
107 IteratorContext* ctx, in MakeDataset()
280 Status Initialize(IteratorContext* ctx) override { in Initialize()
291 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
315 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
340 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
395 void CallCompleted(const std::shared_ptr<IteratorContext>& ctx, in CallCompleted()
411 void CallFunction(std::shared_ptr<IteratorContext> ctx, in CallFunction()
513 void EnsureRunnerThreadStarted(IteratorContext* ctx) in EnsureRunnerThreadStarted()
[all …]
|
| D | assert_next_dataset_op.cc | 100 Status Initialize(IteratorContext* ctx) override { in Initialize()
120 Status GetNextInternal(IteratorContext* ctx, in GetNextInternal()
128 IteratorContext* ctx, model::Node::Args args) const override { in CreateNode()
138 Status RestoreInternal(IteratorContext* ctx, in RestoreInternal()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.h | 279 class IteratorContext {
282 explicit Params(IteratorContext* ctx) in Params()
350 explicit IteratorContext(IteratorContext* ctx) : params_(Params{ctx}) {} in IteratorContext() function
352 explicit IteratorContext(OpKernelContext* ctx) : params_(Params{ctx}) {} in IteratorContext() function
354 explicit IteratorContext(Params params) : params_(std::move(params)) {} in IteratorContext() function
460 virtual Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
463 Status GetNext(IteratorContext&& ctx, std::vector<Tensor>* out_tensors, in GetNext()
484 virtual Status Initialize(IteratorContext* ctx) { return Status::OK(); } in Initialize()
492 virtual Status Restore(IteratorContext* ctx, IteratorStateReader* reader) { in Restore()
499 IteratorContext* ctx, model::Node::Args args) const = 0;
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.cc | 39 Iterator::Iterator(IteratorBase* iterator, IteratorContext* ctx) in Iterator()
87 std::unique_ptr<IteratorContext> ctx; in MakeIterator()
97 IteratorContext::Params params(&op_ctx); in MakeIterator()
99 ctx = MakeUnique<IteratorContext>(std::move(params)); in MakeIterator()
|