| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 132 IteratorContext* ctx, std::unique_ptr<InstantiatedCapturedFunction>* in Instantiate()
157 absl::WrapUnique<InstantiatedCapturedFunction>( in Instantiate()
158 new InstantiatedCapturedFunction(lib, f_handle, std::move(ret_types), in Instantiate()
283 InstantiatedCapturedFunction::InstantiatedCapturedFunction( in InstantiatedCapturedFunction() function in tensorflow::data::InstantiatedCapturedFunction
298 InstantiatedCapturedFunction::~InstantiatedCapturedFunction() {} in ~InstantiatedCapturedFunction()
300 Status InstantiatedCapturedFunction::Run(IteratorContext* ctx, in Run()
304 f_opts.step_id = InstantiatedCapturedFunction::generate_step_id(); in Run()
336 Status InstantiatedCapturedFunction::RunWithBorrowedArgs( in RunWithBorrowedArgs()
340 f_opts.step_id = InstantiatedCapturedFunction::generate_step_id(); in RunWithBorrowedArgs()
373 Status InstantiatedCapturedFunction::RunInstantiated( in RunInstantiated()
[all …]
|
| D | captured_function.h | 49 class InstantiatedCapturedFunction {
51 ~InstantiatedCapturedFunction();
95 InstantiatedCapturedFunction(
109 TF_DISALLOW_COPY_AND_ASSIGN(InstantiatedCapturedFunction);
145 std::unique_ptr<InstantiatedCapturedFunction>*
|
| D | filter_dataset_op.cc | 37 std::function<Status(IteratorContext*, InstantiatedCapturedFunction*,
59 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
77 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
285 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | map_dataset_op.cc | 33 std::function<Status(IteratorContext*, InstantiatedCapturedFunction*,
58 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
68 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
260 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | generator_dataset_op.cc | 143 std::unique_ptr<InstantiatedCapturedFunction> instantiated_init_func_;
144 std::unique_ptr<InstantiatedCapturedFunction> instantiated_next_func_;
145 std::unique_ptr<InstantiatedCapturedFunction> instantiated_finalize_func_;
|
| D | dataset_utils.h | 47 int64 thread_index, const InstantiatedCapturedFunction& inst_captured_func,
|
| D | flat_map_dataset_op.cc | 261 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | parallel_map_dataset_op.cc | 270 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | dataset_utils.cc | 85 int64 thread_index, const InstantiatedCapturedFunction& inst_captured_func, in MakeIteratorFromInputElement()
|
| D | interleave_dataset_op.cc | 316 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | parallel_interleave_dataset_op.cc | 918 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | iterator_ops.cc | 678 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func; in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | take_while_dataset_op.cc | 36 std::function<Status(IteratorContext*, InstantiatedCapturedFunction*,
58 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
76 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
232 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | group_by_reducer_dataset_op.cc | 388 std::unique_ptr<InstantiatedCapturedFunction> instantiated_key_func_;
389 std::unique_ptr<InstantiatedCapturedFunction> instantiated_init_func_;
390 std::unique_ptr<InstantiatedCapturedFunction> instantiated_reduce_func_;
391 std::unique_ptr<InstantiatedCapturedFunction> instantiated_finalize_func_;
|
| D | group_by_window_dataset_op.cc | 487 std::unique_ptr<InstantiatedCapturedFunction> instantiated_key_func_;
488 std::unique_ptr<InstantiatedCapturedFunction> instantiated_reduce_func_;
489 std::unique_ptr<InstantiatedCapturedFunction>
|
| D | map_and_batch_dataset_op.cc | 52 std::function<void(IteratorContext*, InstantiatedCapturedFunction*,
96 InstantiatedCapturedFunction* instantiated_captured_func, in MakeDataset()
108 InstantiatedCapturedFunction* instantiated_captured_func, in MakeDataset()
807 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | scan_dataset_op.cc | 282 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | choose_fastest_branch_dataset_op.cc | 518 std::vector<std::unique_ptr<InstantiatedCapturedFunction>>
|
| D | parallel_interleave_dataset_op.cc | 1040 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|
| D | numa_map_and_batch_dataset_op.cc | 1122 std::unique_ptr<InstantiatedCapturedFunction> instantiated_captured_func_;
|