Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/data/
Dconcatenate_dataset_op.cc99 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph)); in AsGraphDefInternal()
102 b->AddInputDataset(ctx, to_concatenate_, &to_concatenate_graph)); in AsGraphDefInternal()
Dgraph_rewrite_dataset.cc43 db.AddInputDataset(&serialization_ctx, input_, &input_node)); in Optimize()
96 b->AddInputDataset(&optimized_ctx, optimized_input_, output)); in AsGraphDefInternal()
Ddataset_ops.cc40 ctx, db.AddInputDataset(&serialization_ctx, dataset, &input_node)); in Compute()
Dflat_map_dataset_op.cc90 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
101 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in AsGraphDefInternal()
Dfilter_dataset_op.cc132 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
143 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in AsGraphDefInternal()
Dparallel_map_dataset_op.cc142 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
154 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in AsGraphDefInternal()
Dmap_dataset_op.cc141 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
152 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in AsGraphDefInternal()
Dinterleave_dataset_op.cc113 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
127 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in AsGraphDefInternal()
Dshuffle_dataset_op.cc543 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
595 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
682 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dfilter_by_component_dataset_op.cc87 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dtake_dataset_op.cc153 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dshard_dataset_op.cc100 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dzip_dataset_op.cc102 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &input_node)); in AsGraphDefInternal()
Dmodel_dataset_op.cc83 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dskip_dataset_op.cc83 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc180 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
287 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
387 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dtake_while_dataset_op.cc135 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
146 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in AsGraphDefInternal()
Dstats_dataset_ops.cc85 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
195 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
Ddirected_interleave_dataset_op.cc118 b->AddInputDataset(ctx, selector_input_, &selector_input_node)); in AsGraphDefInternal()
122 b->AddInputDataset(ctx, data_inputs_[i], &data_input_nodes[i])); in AsGraphDefInternal()
Dscan_dataset_op.cc108 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
125 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in AsGraphDefInternal()
Dsleep_dataset_op.cc77 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dignore_errors_dataset_op.cc70 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dassert_next_dataset_op.cc86 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
Dgroup_by_reducer_dataset_op.cc117 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
420 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in OtherArgumentsNodeAndType()
Dchoose_fastest_branch_dataset_op.cc244 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
269 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input, &node)); in AsGraphDefInternal()

12