Searched refs:iter_ctx (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | to_tf_record_op.cc | 74 IteratorContext iter_ctx(std::move(params)); in ComputeAsync() local
78 dataset->MakeIterator(&iter_ctx, "ToTFRecordOpIterator", &iterator), in ComputeAsync()
86 ctx, iterator->GetNext(&iter_ctx, &components, &end_of_sequence), in ComputeAsync()
|
| D | set_stats_aggregator_dataset_op.cc | 174 IteratorContext iter_ctx(std::move(params)); in GetNextInternal() local
175 return input_impl_->GetNext(&iter_ctx, out_tensors, end_of_sequence); in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 189 IteratorContext iter_ctx(std::move(params)); in Restore() local
190 TF_RETURN_IF_ERROR(new_state->iterator->Restore(&iter_ctx, reader)); in Restore()
586 IteratorContext iter_ctx(std::move(params)); in ComputeAsync() local
590 dataset->MakeIterator(&iter_ctx, "SingleElementIterator", &iterator), in ComputeAsync()
605 raw_iterator->GetNext(&iter_ctx, &components, &end_of_sequence); in ComputeAsync()
621 raw_iterator->GetNext(&iter_ctx, &components, &end_of_sequence); in ComputeAsync()
677 IteratorContext iter_ctx(std::move(params)); in ComputeAsync() local
681 captured_func->Instantiate(&iter_ctx, &instantiated_captured_func), in ComputeAsync()
686 ctx, dataset->MakeIterator(&iter_ctx, "ReduceIterator", &iterator), in ComputeAsync()
702 status = raw_iterator->GetNext(&iter_ctx, &next_input_element, in ComputeAsync()
[all …]
|
| D | multi_device_iterator_ops.cc | 103 IteratorContext iter_ctx(std::move(params)); in GetNextFromShard() local
104 multi_device_buffer_->GetNextFromShard(&iter_ctx, shard_num, incarnation_id, in GetNextFromShard()
511 IteratorContext iter_ctx(std::move(params)); in Compute() local
513 ctx, dataset->MakeIterator(std::move(iter_ctx), "Iterator", &iterator)); in Compute()
|