Searched refs:iter_ctx (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | set_stats_aggregator_dataset_op.cc | 176 IteratorContext iter_ctx = ContextWithAggregator(ctx); in Initialize() local
177 return dataset()->input_->MakeIterator(&iter_ctx, this, prefix(), in Initialize()
185 IteratorContext iter_ctx = ContextWithAggregator(ctx); in GetNextInternal() local
186 return input_impl_->GetNext(&iter_ctx, out_tensors, end_of_sequence); in GetNextInternal()
197 IteratorContext iter_ctx(std::move(params)); in ContextWithAggregator() local
198 return iter_ctx; in ContextWithAggregator()
|
| D | to_tf_record_op.cc | 85 IteratorContext iter_ctx(std::move(params)); in DoCompute() local
88 &iter_ctx, /*parent=*/nullptr, "ToTFRecordOpIterator", &iterator)); in DoCompute()
95 iterator->GetNext(&iter_ctx, &components, &end_of_sequence)); in DoCompute()
|
| D | io_ops.cc | 85 IteratorContext iter_ctx(std::move(params)); in WriteData() local
88 captured_func->Instantiate(&iter_ctx, &instantiated_captured_func)); in WriteData()
92 dataset->MakeIterator(&iter_ctx, /*parent=*/nullptr, "Save", &iterator)); in WriteData()
104 TF_RETURN_IF_ERROR(iterator->GetNext(&iter_ctx, &element, &end_of_input)); in WriteData()
113 &iter_ctx, instantiated_captured_func.get(), element, &shard_index)); in WriteData()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 643 IteratorContext iter_ctx(std::move(params)); in DoCompute() local
646 &iter_ctx, /*parent=*/nullptr, "SingleElementIterator", &iterator)); in DoCompute()
653 iterator->GetNext(&iter_ctx, &components, &end_of_sequence)); in DoCompute()
666 iterator->GetNext(&iter_ctx, &components, &end_of_sequence)); in DoCompute()
715 IteratorContext iter_ctx(std::move(params)); in DoCompute() local
718 captured_func->Instantiate(&iter_ctx, &instantiated_captured_func)); in DoCompute()
721 TF_RETURN_IF_ERROR(dataset->MakeIterator(&iter_ctx, /*parent=*/nullptr, in DoCompute()
732 iterator->GetNext(&iter_ctx, &next_input_element, &end_of_input)); in DoCompute()
746 &iter_ctx, std::move(args), &reduce_func_output, /*node=*/nullptr)); in DoCompute()
|
| D | multi_device_iterator_ops.cc | 117 IteratorContext iter_ctx(std::move(params)); in GetNextFromShard() local
125 multi_device_buffer_->GetNextFromShard(&iter_ctx, shard_num, incarnation_id, in GetNextFromShard()
576 IteratorContext iter_ctx(std::move(params)); in Compute() local
578 ctx, dataset->MakeIterator(std::move(iter_ctx), /*parent=*/nullptr, in Compute()
|