| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | flat_map_dataset_op.cc | 122 ctx, &instantiated_captured_func_); in Initialize()
250 *instantiated_captured_func_, prefix(), ¤t_element_iterator_, in BuildCurrentElementIteratorLocked()
256 *instantiated_captured_func_, prefix(), ¤t_element_iterator_, in BuildCurrentElementIteratorLocked()
266 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::FlatMapDatasetOp::Dataset::Iterator
|
| D | map_dataset_op.cc | 144 ctx, &instantiated_captured_func_); in Initialize()
161 Status s = instantiated_captured_func_->Run(ctx, std::move(args), in GetNextInternal()
204 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::MapDatasetOp::Dataset::Iterator
|
| D | filter_dataset_op.cc | 121 ctx, &instantiated_captured_func_); in Initialize()
150 TF_RETURN_IF_ERROR(instantiated_captured_func_->RunWithBorrowedArgs( in GetNextInternal()
238 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::FilterDatasetOp::Dataset::Iterator
|
| D | interleave_dataset_op.cc | 138 ctx, &instantiated_captured_func_); in Initialize()
184 *instantiated_captured_func_, prefix(), in GetNextInternal()
281 ctx, this, args_list_[idx], idx, *instantiated_captured_func_, in RestoreCurrentElements()
300 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::InterleaveDatasetOp::Dataset::Iterator
|
| D | parallel_map_dataset_op.cc | 236 ctx, &instantiated_captured_func_); in Initialize()
443 instantiated_captured_func_->RunAsync( in CallFunction()
452 return instantiated_captured_func_->Run( in CallFunction()
650 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::ParallelMapDatasetOp::Dataset::Iterator
|
| D | parallel_interleave_dataset_op.cc | 353 ctx, &instantiated_captured_func_); in Initialize()
1010 *instantiated_captured_func_, prefix(), &element->iterator, in InitializeInputs()
1304 *instantiated_captured_func_.get(), prefix(), &iterator, in ReadElement()
1547 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::ParallelInterleaveDatasetOp::Dataset::ParallelInterleaveIterator
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | take_while_dataset_op.cc | 132 ctx, &instantiated_captured_func_); in Initialize()
153 TF_RETURN_IF_ERROR(instantiated_captured_func_->RunWithBorrowedArgs( in GetNextInternal()
201 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::experimental::__anonaa7f23910111::TakeWhileDatasetOp::Dataset::Iterator
|
| D | scan_dataset_op.cc | 178 ctx, &instantiated_captured_func_); in Initialize()
203 Status s = instantiated_captured_func_->Run( in GetNextInternal()
301 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::experimental::__anon700576d90111::ScanDatasetOp::Dataset::Iterator
|
| D | io_ops.cc | 261 ctx, &instantiated_captured_func_)); in Initialize()
319 TF_RETURN_IF_ERROR(instantiated_captured_func_->Run( in InitializeInput()
336 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::experimental::LoadDatasetOp::Dataset::Iterator
|
| D | map_and_batch_dataset_op.cc | 227 ctx, &instantiated_captured_func_); in Initialize()
472 instantiated_captured_func_->RunAsync(ctx.get(), std::move(input_element), in CallFunction()
674 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::experimental::MapAndBatchDatasetOp::Dataset::Iterator
|
| D | parallel_interleave_dataset_op.cc | 290 ctx, &instantiated_captured_func_); in Initialize()
775 thread_index, *instantiated_captured_func_, prefix(), in WorkerThread()
1017 ctx, this, state->input, index, *instantiated_captured_func_, in ReadWorkerThreadStateLocked()
1131 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_; member in tensorflow::data::experimental::ParallelInterleaveDatasetOp::Dataset::Iterator
|