| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_dataset_op.cc | 49 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
52 &captured_func)); in MakeDataset()
55 CapturedFunction* raw_captured_func = captured_func.get(); in MakeDataset()
90 new Dataset(ctx, input, func_, std::move(captured_func), output_types_, in MakeDataset()
100 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
110 captured_func_(std::move(captured_func)), in Dataset()
|
| D | flat_map_dataset_op.cc | 42 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
44 &captured_func)); in MakeDataset()
45 *output = new Dataset(ctx, input, func_, std::move(captured_func), in MakeDataset()
54 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
60 captured_func_(std::move(captured_func)), in Dataset()
|
| D | filter_dataset_op.cc | 52 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
54 &captured_func)); in MakeDataset()
89 *output = new Dataset(ctx, input, func_, std::move(captured_func), in MakeDataset()
98 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
103 captured_func_(std::move(captured_func)), in Dataset()
|
| D | parallel_map_dataset_op.cc | 64 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
67 &captured_func)); in MakeDataset()
75 std::move(captured_func), short_circuit_indices_, in MakeDataset()
87 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
98 captured_func_(std::move(captured_func)), in Dataset()
|
| D | interleave_dataset_op.cc | 60 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
62 &captured_func)); in MakeDataset()
65 new Dataset(ctx, input, func_, std::move(captured_func), cycle_length, in MakeDataset()
74 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() argument
80 captured_func_(std::move(captured_func)), in Dataset()
|
| D | parallel_interleave_dataset_op.cc | 89 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
92 &captured_func)); in MakeDataset()
99 new Dataset(ctx, input, interleave_func_, std::move(captured_func), in MakeDataset()
109 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() argument
116 captured_func_(std::move(captured_func)), in Dataset()
|
| D | captured_function.h | 99 CapturedFunction* captured_func);
|
| D | captured_function.cc | 286 CapturedFunction* captured_func) in InstantiatedCapturedFunction() argument
291 captured_func_(captured_func) {} in InstantiatedCapturedFunction()
|
| D | iterator_ops.cc | 663 std::unique_ptr<CapturedFunction> captured_func; in ComputeAsync() local
667 use_inter_op_parallelism_, &captured_func), in ComputeAsync()
681 captured_func->Instantiate(&iter_ctx, &instantiated_captured_func), in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | take_while_dataset_op.cc | 51 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
53 &captured_func)); in MakeDataset()
87 *output = new Dataset(ctx, input, func_, std::move(captured_func), in MakeDataset()
96 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
101 captured_func_(std::move(captured_func)), in Dataset()
|
| D | scan_dataset_op.cc | 52 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
54 &captured_func)); in MakeDataset()
57 std::move(captured_func), state_types_, output_types_, in MakeDataset()
66 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
75 captured_func_(std::move(captured_func)), in Dataset()
|
| D | group_by_reducer_dataset_op.cc | 410 const std::unique_ptr<CapturedFunction>& captured_func, in OtherArgumentsNodeAndType() argument
413 other_arguments_node->reserve(captured_func->captured_inputs().size()); in OtherArgumentsNodeAndType()
414 other_arguments_types->reserve(captured_func->captured_inputs().size()); in OtherArgumentsNodeAndType()
415 for (const Tensor& t : captured_func->captured_inputs()) { in OtherArgumentsNodeAndType()
|
| D | map_and_batch_dataset_op.cc | 88 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
90 &captured_func)); in MakeDataset()
93 CapturedFunction* raw_captured_func = captured_func.get(); in MakeDataset()
139 std::move(captured_func), &ctx->eigen_cpu_device(), in MakeDataset()
151 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
162 captured_func_(std::move(captured_func)), in Dataset()
|
| D | group_by_window_dataset_op.cc | 495 const std::unique_ptr<CapturedFunction>& captured_func, in OtherArgumentsNodeAndType() argument
498 other_arguments_node->reserve(captured_func->captured_inputs().size()); in OtherArgumentsNodeAndType()
499 other_arguments_types->reserve(captured_func->captured_inputs().size()); in OtherArgumentsNodeAndType()
500 for (const Tensor& t : captured_func->captured_inputs()) { in OtherArgumentsNodeAndType()
|
| D | parallel_interleave_dataset_op.cc | 77 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
80 &captured_func)); in MakeDataset()
83 new Dataset(ctx, input, interleave_func_, std::move(captured_func), in MakeDataset()
93 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() argument
100 captured_func_(std::move(captured_func)), in Dataset()
|
| D | numa_map_and_batch_dataset_op.cc | 87 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
91 &captured_func)); in MakeDataset()
95 std::move(captured_func)); in MakeDataset()
106 std::unique_ptr<CapturedFunction> captured_func) in Dataset() argument
115 captured_func_(std::move(captured_func)) { in Dataset()
|