Home
last modified time | relevance | path

Searched refs:AddInputDataset (Results 1 – 25 of 54) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dshuffle_dataset_op.cc546 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
606 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
661 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
852 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
910 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dconcatenate_dataset_op.cc143 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph)); in AsGraphDefInternal()
146 b->AddInputDataset(ctx, to_concatenate_, &to_concatenate_graph)); in AsGraphDefInternal()
H A Dcache_dataset_ops.cc689 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph)); in AsGraphDefInternal()
710 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
1049 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
1094 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
H A Dparallel_interleave_dataset_op_test.cc656 db.AddInputDataset(&serialization_ctx, dataset->dataset(), &output_node)); in TEST_F()
697 db.AddInputDataset(&serialization_ctx, dataset->dataset(), &output_node)); in TEST_F()
H A Doptions_dataset_op.cc99 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dtake_dataset_op.cc219 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dskip_dataset_op.cc106 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dfilter_dataset_op.cc93 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dmap_dataset_op.cc120 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dzip_dataset_op.cc153 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &input_node)); in AsGraphDefInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dthreadpool_dataset_op.cc218 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
338 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
479 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dstats_dataset_ops.cc98 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
217 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
H A Ddirected_interleave_dataset_op.cc133 b->AddInputDataset(ctx, selector_input_, &selector_input_node)); in AsGraphDefInternal()
137 b->AddInputDataset(ctx, data_inputs_[i], &data_input_nodes[i])); in AsGraphDefInternal()
H A Drebatch_dataset_op.cc109 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
370 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dignore_errors_dataset_op.cc84 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dsleep_dataset_op.cc88 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dassert_next_dataset_op.cc81 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dassert_cardinality_dataset_op.cc85 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dtake_while_dataset_op.cc101 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
H A Dlookup_ops.cc127 db.AddInputDataset(&serialization_ctx, dataset, &dataset_node)); in MakeDatasetInitializerSerializer()
H A Dunique_dataset_op.cc71 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dsampling_dataset_op.cc87 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dset_stats_aggregator_dataset_op.cc158 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dassert_prev_dataset_op.cc180 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
H A Dunbatch_dataset_op.cc107 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()

123