Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc99 std::vector<const Edge*> data_edges, control_edges; in rewriteNode() local
104 data_edges.push_back(input_edge); in rewriteNode()
130 .Input(data_edges[0]->src(), data_edges[0]->src_output()) in rewriteNode()
147 for (int i = 0; i < data_edges.size(); ++i) { in rewriteNode()
153 .Input(data_edges[i]->src(), in rewriteNode()
154 data_edges[i]->src_output()) // value: T in rewriteNode()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc223 std::vector<OutEdgeInfo> data_edges; in PostprocessDataEdgesBetweenOutsideCompilations() local
228 data_edges.push_back({e->dst(), e->src_output(), e->dst_input()}); in PostprocessDataEdgesBetweenOutsideCompilations()
235 for (int i = 0; i < data_edges.size(); i++) { in PostprocessDataEdgesBetweenOutsideCompilations()
236 Node* dst = data_edges[i].dst; in PostprocessDataEdgesBetweenOutsideCompilations()
238 int dst_input = data_edges[i].dst_input; in PostprocessDataEdgesBetweenOutsideCompilations()
250 for (int j = i + 1; j < data_edges.size(); j++) { in PostprocessDataEdgesBetweenOutsideCompilations()
251 if (data_edges[j].dst == dst) { in PostprocessDataEdgesBetweenOutsideCompilations()
252 data_edges[j].dst = replace_node; in PostprocessDataEdgesBetweenOutsideCompilations()
Dbuild_xla_ops_pass.cc84 std::vector<const Edge*> data_edges; in MergeOutgoingDataEdges() local
85 absl::c_copy_if(old_node->out_edges(), std::back_inserter(data_edges), in MergeOutgoingDataEdges()
88 for (const Edge* e : data_edges) { in MergeOutgoingDataEdges()