| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | io_ops.cc | 59 std::unique_ptr<CapturedFunction> captured_func; in DoCompute() local
61 ctx, func_metadata_, kShardFuncOtherArgs, &captured_func)); in DoCompute()
64 TF_RETURN_IF_ERROR(WriteData(ctx, dataset, std::move(captured_func), run_dir, in DoCompute()
73 std::unique_ptr<CapturedFunction> captured_func, in WriteData() argument
88 captured_func->Instantiate(&iter_ctx, &instantiated_captured_func)); in WriteData()
182 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
186 captured_func_(std::move(captured_func)), in Dataset()
359 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
362 &captured_func)); in MakeDataset()
374 std::move(captured_func), output_types_, output_shapes_); in MakeDataset()
|
| D | take_while_dataset_op.cc | 45 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
48 &captured_func)); in MakeDataset()
49 *output = new Dataset(ctx, input, std::move(captured_func)); in MakeDataset()
56 std::unique_ptr<CapturedFunction> captured_func) in Dataset() argument
59 captured_func_(std::move(captured_func)) { in Dataset()
|
| D | scan_dataset_op.cc | 59 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
62 &captured_func)); in MakeDataset()
66 std::move(captured_func), state_types_, output_types_, in MakeDataset()
75 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
83 captured_func_(std::move(captured_func)), in Dataset()
|
| D | choose_fastest_branch_dataset_op.cc | 259 for (const auto& captured_func : captured_funcs_) { in CheckExternalState() local
260 TF_RETURN_IF_ERROR(captured_func->CheckExternalState()); in CheckExternalState()
289 for (const auto& captured_func : captured_funcs_) { in AsGraphDefInternal() local
290 TF_RETURN_IF_ERROR(captured_func->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
| D | map_and_batch_dataset_op.cc | 88 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
97 captured_func_(std::move(captured_func)), in Dataset()
719 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
722 &captured_func)); in MakeDataset()
730 std::move(captured_func), preserve_cardinality_); in MakeDataset()
|
| D | io_ops.h | 53 std::unique_ptr<CapturedFunction> captured_func,
|
| D | parallel_interleave_dataset_op.cc | 96 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() argument
103 captured_func_(std::move(captured_func)), in Dataset()
1226 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
1229 &captured_func)); in MakeDataset()
1231 *output = new Dataset(ctx, input, std::move(captured_func), cycle_length, in MakeDataset()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_dataset_op.cc | 44 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
51 captured_func_(std::move(captured_func)), in Dataset()
229 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
232 &captured_func)); in MakeDataset()
234 *output = new Dataset(ctx, input, std::move(captured_func), output_types_, in MakeDataset()
|
| D | filter_dataset_op.cc | 51 std::unique_ptr<CapturedFunction> captured_func) in Dataset() argument
54 captured_func_(std::move(captured_func)) { in Dataset()
256 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
259 &captured_func)); in MakeDataset()
261 *output = new Dataset(ctx, input, std::move(captured_func)); in MakeDataset()
|
| D | flat_map_dataset_op.cc | 49 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
54 captured_func_(std::move(captured_func)), in Dataset()
285 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
288 &captured_func)); in MakeDataset()
289 *output = new Dataset(ctx, input, std::move(captured_func), output_types_, in MakeDataset()
|
| D | interleave_dataset_op.cc | 54 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() argument
59 captured_func_(std::move(captured_func)), in Dataset()
337 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
340 &captured_func)); in MakeDataset()
342 *output = new Dataset(ctx, input, std::move(captured_func), cycle_length, in MakeDataset()
|
| D | captured_function.h | 218 CapturedFunction* captured_func, bool is_multi_device,
277 CapturedFunction* captured_func, bool is_multi_device);
|
| D | parallel_map_dataset_op.cc | 78 std::unique_ptr<CapturedFunction> captured_func, in Dataset() argument
87 captured_func_(std::move(captured_func)), in Dataset()
721 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
724 &captured_func)); in MakeDataset()
732 deterministic_, std::move(captured_func), in MakeDataset()
|
| D | parallel_interleave_dataset_op.cc | 162 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() argument
169 captured_func_(std::move(captured_func)), in Dataset()
1655 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
1658 &captured_func)); in MakeDataset()
1665 ctx, input, std::move(captured_func), cycle_length, block_length, in MakeDataset()
|
| D | captured_function.cc | 809 CapturedFunction* captured_func, bool is_multi_device, in Create() argument
812 lib, f_handle, ret_types, runner, captured_func, is_multi_device)); in Create()
819 CapturedFunction* captured_func, bool is_multi_device) in InstantiatedCapturedFunction() argument
824 captured_func_(captured_func), in InstantiatedCapturedFunction()
|
| D | iterator_ops.cc | 702 std::unique_ptr<CapturedFunction> captured_func; in DoCompute() local
704 ctx, func_metadata_, "other_arguments", &captured_func)); in DoCompute()
718 captured_func->Instantiate(&iter_ctx, &instantiated_captured_func)); in DoCompute()
|