Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc190 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
191 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
195 this, {input_graph_node, resource_handle_node}, output)); in AsGraphDefInternal()
321 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
322 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
327 this, {input_graph_node, max_intra_op_parallelism_node}, output)); in AsGraphDefInternal()
446 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
447 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
451 b->AddDataset(this, {input_graph_node, num_threads_node}, output)); in AsGraphDefInternal()
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()
Dsleep_dataset_op.cc85 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
86 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
93 {{0, input_graph_node}, in AsGraphDefInternal()
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()
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()
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()
Drebatch_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()
107 b->AddDataset(this, {input_graph_node, num_replicas}, output)); in AsGraphDefInternal()
357 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
358 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
364 this, {input_graph_node, batch_sizes, drop_remainder}, output)); in AsGraphDefInternal()
Dset_stats_aggregator_dataset_op.cc155 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
156 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
164 this, {input_graph_node, resource_handle_node, tag_node, prefix_node}, in AsGraphDefInternal()
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()
Dunbatch_dataset_op.cc101 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
102 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
103 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
Dsliding_window_dataset_op.cc125 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
126 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
134 this, {input_graph_node, window_size, window_shift, window_stride}, in AsGraphDefInternal()
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()
Dgroup_by_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()
174 this, {{0, input_graph_node}}, in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.cc467 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
468 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
481 {input_graph_node, buffer_size_node, seed_node, seed2_node}, // Inputs in AsGraphDefInternal()
527 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
528 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
537 {input_graph_node, buffer_size_node, resource_handle_node}, // Inputs in AsGraphDefInternal()
582 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
583 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
599 {input_graph_node, buffer_size_node, seed_node, in AsGraphDefInternal()
773 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
[all …]
Dmodel_dataset_op.cc99 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
100 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
101 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
110 b->AddDataset(this, {input_graph_node}, in AsGraphDefInternal()
Dskip_dataset_op.cc91 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
92 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
95 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dtake_dataset_op.cc198 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
199 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
202 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dmap_dataset_op.cc97 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
98 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
123 this, {std::make_pair(0, input_graph_node)}, // Single tensor inputs. in AsGraphDefInternal()
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()
Dshard_dataset_op.cc100 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
101 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
111 b->AddDataset(this, {input_graph_node, num_shards, index}, in AsGraphDefInternal()
Dbatch_dataset_op.cc133 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
134 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
142 b->AddDataset(this, {input_graph_node, batch_size, drop_remainder}, in AsGraphDefInternal()
Drepeat_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()
Dflat_map_dataset_op.cc92 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
93 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
104 this, {std::make_pair(0, input_graph_node)}, // Single tensor inputs. in AsGraphDefInternal()
Dpadded_batch_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()
173 this, {{0, input_graph_node}, {1, batch_size}, {4, drop_remainder}}, in AsGraphDefInternal()
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()

12