Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/data/kernels/
Dignore_errors_dataset_op.cc65 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
66 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
67 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/kernels/data/
Dtake_dataset_op.cc77 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
78 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
82 b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dskip_dataset_op.cc76 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
77 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
81 b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dfilter_dataset_op.cc90 Node* input_graph_node; in AsGraphDefInternal() local
91 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
109 this, {{0, input_graph_node}}, {{1, other_arguments}}, in AsGraphDefInternal()
Dunique_dataset_op.cc80 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
81 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
82 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
Dmap_dataset_op.cc95 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
96 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
114 this, {std::make_pair(0, input_graph_node)}, // Single tensor inputs. in AsGraphDefInternal()
Dbatch_dataset_op.cc85 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
86 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
90 b->AddDataset(this, {input_graph_node, batch_size}, output)); in AsGraphDefInternal()
Drepeat_dataset_op.cc77 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
78 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
82 b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dshuffle_dataset_op.cc411 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
412 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
423 this, {input_graph_node, buffer_size, seed, seed2}, // Inputs in AsGraphDefInternal()
495 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
496 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
507 this, {input_graph_node, buffer_size, seed, seed2, count}, // Inputs in AsGraphDefInternal()
Dflat_map_dataset_op.cc97 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
98 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
116 this, {std::make_pair(0, input_graph_node)}, // Single tensor inputs. in AsGraphDefInternal()
Dprefetch_dataset_op.cc74 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
75 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
79 b->AddDataset(this, {input_graph_node, buffer_size}, output)); in AsGraphDefInternal()
Dpadded_batch_dataset_op.cc144 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
145 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
176 b->AddDataset(this, {{0, input_graph_node}, {1, batch_size}}, in AsGraphDefInternal()
Dparallel_map_dataset_op.cc108 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
109 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
139 {std::make_pair(0, input_graph_node), in AsGraphDefInternal()
Dgroup_by_window_dataset_op.cc142 Node* input_graph_node = nullptr; in AsGraphDefInternal() local
143 TF_RETURN_IF_ERROR(b->AddParentDataset(ctx, input_, &input_graph_node)); in AsGraphDefInternal()
181 this, {{0, input_graph_node}}, in AsGraphDefInternal()