Home
last modified time | relevance | path

Searched refs:input_graph_node (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dshuffle_dataset_op.cc545 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
546 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
559 {input_graph_node, buffer_size_node, seed_node, seed2_node}, // Inputs in AsGraphDefInternal()
605 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
606 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
615 {input_graph_node, buffer_size_node, resource_handle_node}, // Inputs in AsGraphDefInternal()
660 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
661 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
677 {input_graph_node, buffer_size_node, seed_node, in AsGraphDefInternal()
851 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
[all …]
H A Doptions_dataset_op.cc98 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
99 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
103 this, {input_graph_node}, in AsGraphDefInternal()
H A Dtake_dataset_op.cc218 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
219 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
222 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
H A Dskip_dataset_op.cc105 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
106 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
109 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
H A Dfilter_dataset_op.cc92 Node* input_graph_node; in AsGraphDefInternal() local
93 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
104 this, {{0, input_graph_node}}, {{1, other_arguments}}, in AsGraphDefInternal()
H A Dmap_dataset_op.cc119 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
120 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
145 this, {std::make_pair(0, input_graph_node)}, // Single tensor inputs. in AsGraphDefInternal()
H A Dmodel_dataset_op.cc106 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
107 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
116 b->AddDataset(this, {input_graph_node}, in AsGraphDefInternal()
H A Dshard_dataset_op.cc116 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
117 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
127 b->AddDataset(this, {input_graph_node, num_shards, index}, in AsGraphDefInternal()
H A Dbatch_dataset_op.cc163 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
164 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
172 b->AddDataset(this, {input_graph_node, batch_size, drop_remainder}, in AsGraphDefInternal()
H A Drepeat_dataset_op.cc130 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
131 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
134 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
H A Dflat_map_dataset_op.cc102 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
103 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
114 this, {std::make_pair(0, input_graph_node)}, // Single tensor inputs. in AsGraphDefInternal()
H A Dwindow_dataset_op.cc121 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
122 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
133 {input_graph_node, window_size_node, window_shift_node, in AsGraphDefInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dthreadpool_dataset_op.cc217 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
218 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
222 this, {input_graph_node, resource_handle_node}, output)); in AsGraphDefInternal()
337 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
338 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
343 this, {input_graph_node, max_intra_op_parallelism_node}, output)); in AsGraphDefInternal()
478 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
479 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
483 b->AddDataset(this, {input_graph_node, num_threads_node}, output)); in AsGraphDefInternal()
H A Drebatch_dataset_op.cc108 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
109 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
113 b->AddDataset(this, {input_graph_node, num_replicas}, output)); in AsGraphDefInternal()
369 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
370 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
376 this, {input_graph_node, batch_sizes, drop_remainder}, output)); in AsGraphDefInternal()
H A Dignore_errors_dataset_op.cc83 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
84 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
88 b->AddDataset(this, {std::make_pair(0, input_graph_node)}, {}, in AsGraphDefInternal()
H A Dsleep_dataset_op.cc87 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
88 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
95 {{0, input_graph_node}, in AsGraphDefInternal()
H A Dassert_next_dataset_op.cc80 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
81 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
85 b->AddDataset(this, {input_graph_node, transformations_node}, output)); in AsGraphDefInternal()
H A Dassert_cardinality_dataset_op.cc84 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
85 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
89 b->AddDataset(this, {input_graph_node, cardinality_node}, output)); in AsGraphDefInternal()
H A Dunique_dataset_op.cc70 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
71 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
72 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
H A Dsampling_dataset_op.cc86 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
87 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
95 b->AddDataset(this, {input_graph_node, rate, seed, seed2}, output)); in AsGraphDefInternal()
H A Dset_stats_aggregator_dataset_op.cc157 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
158 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
166 this, {input_graph_node, resource_handle_node, tag_node, prefix_node}, in AsGraphDefInternal()
H A Dassert_prev_dataset_op.cc179 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
180 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
184 b->AddDataset(this, {input_graph_node, transformations_node}, output)); in AsGraphDefInternal()
H A Dunbatch_dataset_op.cc106 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
107 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
108 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
H A Dsliding_window_dataset_op.cc135 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
136 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
149 this, {input_graph_node, window_size, window_shift, window_stride}, in AsGraphDefInternal()
H A Dgroup_by_reducer_dataset_op.cc128 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
129 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
178 this, {{0, input_graph_node}}, in AsGraphDefInternal()

12