Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_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 this, {input_graph_node, resource_handle_node}, output)); in AsGraphDefInternal()
286 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
287 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
292 this, {input_graph_node, max_intra_op_parallelism_node}, output)); in AsGraphDefInternal()
386 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
387 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
391 b->AddDataset(this, {input_graph_node, num_threads_node}, output)); in AsGraphDefInternal()
Dsleep_dataset_op.cc76 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
77 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
84 {{0, input_graph_node}, in AsGraphDefInternal()
Dignore_errors_dataset_op.cc69 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
70 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
71 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
Dassert_next_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()
90 this, {input_graph_node, transformations_node}, output)); in AsGraphDefInternal()
Dunique_dataset_op.cc81 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
82 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
83 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
Dset_stats_aggregator_dataset_op.cc139 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
140 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
148 this, {input_graph_node, resource_handle_node, tag_node, prefix_node}, in AsGraphDefInternal()
Dunbatch_dataset_op.cc78 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
79 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
80 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
Dsliding_window_dataset_op.cc117 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
118 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
126 this, {input_graph_node, window_size, window_shift, window_stride}, in AsGraphDefInternal()
Dparse_example_dataset_op.cc213 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
214 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
245 {0, input_graph_node}, in AsGraphDefInternal()
Dgroup_by_reducer_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()
166 this, {{0, input_graph_node}}, in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.cc542 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
543 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
554 this, {input_graph_node, buffer_size, seed, seed2}, // Inputs in AsGraphDefInternal()
594 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
595 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
606 this, {input_graph_node, buffer_size, seed, seed2}, // Inputs in AsGraphDefInternal()
681 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
682 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
693 this, {input_graph_node, buffer_size, seed, seed2, count}, // Inputs in AsGraphDefInternal()
Dfilter_by_component_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()
90 this, {std::make_pair(0, input_graph_node)}, // Single tensor inputs. in AsGraphDefInternal()
Dtake_dataset_op.cc152 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
153 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
156 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dshard_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()
106 b->AddDataset(this, {input_graph_node, num_shards, index}, output)); in AsGraphDefInternal()
Dmodel_dataset_op.cc82 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
83 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
84 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
Dskip_dataset_op.cc82 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
83 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
87 b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dbatch_dataset_op.cc111 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
112 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
118 this, {input_graph_node, batch_size, drop_remainder}, output)); in AsGraphDefInternal()
Drepeat_dataset_op.cc95 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
96 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
100 b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dfilter_dataset_op.cc131 Node* input_graph_node; in AsGraphDefInternal() local
132 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
156 this, {{0, input_graph_node}}, {{1, other_arguments}}, in AsGraphDefInternal()
Dflat_map_dataset_op.cc89 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
90 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()
Dparallel_map_dataset_op.cc141 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
142 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
191 {std::make_pair(0, input_graph_node), in AsGraphDefInternal()
Dmap_dataset_op.cc140 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
141 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
179 this, {std::make_pair(0, input_graph_node)}, // Single tensor inputs. in AsGraphDefInternal()
Dpadded_batch_dataset_op.cc168 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
169 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
203 this, {{0, input_graph_node}, {1, batch_size}, {4, drop_remainder}}, in AsGraphDefInternal()
Dprefetch_dataset_op.cc69 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
70 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
74 b->AddDataset(this, {input_graph_node, buffer_size}, output)); in AsGraphDefInternal()
Dwindow_dataset_op.cc114 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
115 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
126 {input_graph_node, window_size_node, window_shift_node, in AsGraphDefInternal()

12