/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,
|