Home
last modified time | relevance | path

Searched refs:data_edges (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc131 std::vector<const Edge*> data_edges, control_edges; in RewriteIntoTempVariable() local
136 data_edges.push_back(input_edge); in RewriteIntoTempVariable()
162 .Input(data_edges[0]->src(), data_edges[0]->src_output()) in RewriteIntoTempVariable()
179 for (int i = 0; i < data_edges.size(); ++i) { in RewriteIntoTempVariable()
185 .Input(data_edges[i]->src(), in RewriteIntoTempVariable()
186 data_edges[i]->src_output()) // value: T in RewriteIntoTempVariable()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc229 std::vector<OutEdgeInfo> data_edges; in PostprocessDataEdgesBetweenOutsideCompilations() local
234 data_edges.push_back({e->dst(), e->src_output(), e->dst_input()}); in PostprocessDataEdgesBetweenOutsideCompilations()
241 for (int i = 0, end = data_edges.size(); i < end; i++) { in PostprocessDataEdgesBetweenOutsideCompilations()
242 Node* dst = data_edges[i].dst; in PostprocessDataEdgesBetweenOutsideCompilations()
244 int dst_input = data_edges[i].dst_input; in PostprocessDataEdgesBetweenOutsideCompilations()
256 for (int j = i + 1, end = data_edges.size(); j < end; j++) { in PostprocessDataEdgesBetweenOutsideCompilations()
257 if (data_edges[j].dst == dst) { in PostprocessDataEdgesBetweenOutsideCompilations()
258 data_edges[j].dst = replace_node; in PostprocessDataEdgesBetweenOutsideCompilations()
Dbuild_xla_ops_pass.cc125 std::vector<const Edge*> data_edges; in MergeOutgoingDataEdges() local
126 absl::c_copy_if(old_node->out_edges(), std::back_inserter(data_edges), in MergeOutgoingDataEdges()
129 for (const Edge* e : data_edges) { in MergeOutgoingDataEdges()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc630 std::vector<const Edge*> data_edges; in CreatePlaceholderNodeForFeed() local
634 data_edges.push_back(edge); in CreatePlaceholderNodeForFeed()
640 for (const auto* edge : data_edges) { in CreatePlaceholderNodeForFeed()