Home
last modified time | relevance | path

Searched defs:captured_func (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dio_ops.cc92 std::unique_ptr<CapturedFunction> captured_func; in DoCompute() local
106 std::unique_ptr<CapturedFunction> captured_func, in WriteData()
558 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
747 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
H A Dtake_while_dataset_op.cc45 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
56 std::unique_ptr<CapturedFunction> captured_func) in Dataset()
H A Dscan_dataset_op.cc59 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
75 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
H A Dchoose_fastest_branch_dataset_op.cc259 for (const auto& captured_func : captured_funcs_) { in CheckExternalState() local
289 for (const auto& captured_func : captured_funcs_) { in AsGraphDefInternal() local
H A Dmap_and_batch_dataset_op.cc88 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
745 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
H A Dparallel_interleave_dataset_op.cc98 std::unique_ptr<CapturedFunction> captured_func, int64_t cycle_length, in Dataset()
1271 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dfilter_dataset_op.cc51 std::unique_ptr<CapturedFunction> captured_func) in Dataset()
256 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
H A Dmap_dataset_op.cc45 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
254 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
H A Dflat_map_dataset_op.cc59 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
333 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
H A Dparallel_map_dataset_op.cc81 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
91 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
769 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
H A Dinterleave_dataset_op.cc54 std::unique_ptr<CapturedFunction> captured_func, int64_t cycle_length, in Dataset()
466 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
H A Dparallel_filter_dataset_op.cc65 std::unique_ptr<CapturedFunction> captured_func) in Dataset()
648 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
H A Dreduce_dataset_op.cc59 std::unique_ptr<CapturedFunction> captured_func; in DoCompute() local
H A Dparallel_interleave_dataset_op.cc199 std::unique_ptr<CapturedFunction> captured_func, int64_t cycle_length, in Dataset()
1757 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.cc771 CapturedFunction* captured_func, bool is_multi_device) in InstantiatedCapturedFunction()