Home
last modified time | relevance | path

Searched refs:output_edges (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc135 std::vector<const Edge*> output_edges; in ReorderOutputEdges() local
140 output_edges.push_back(e); in ReorderOutputEdges()
142 for (const Edge* e : output_edges) { in ReorderOutputEdges()
/external/tensorflow/tensorflow/core/common_runtime/
Dsimple_propagator_state.cc77 for (const EdgeInfo& e : item->output_edges()) { in PropagateOutputs()
Dgraph_view.h117 gtl::ArraySlice<EdgeInfo> output_edges() const { in output_edges() function
Dpropagator_state.cc313 for (const EdgeInfo& e : item->output_edges()) { in DeleteFrame()
414 for (const EdgeInfo& e : item->output_edges()) { in ActivateNodesFastPathInternal()
471 for (const EdgeInfo& e : item->output_edges()) { in ActivateNodesSlowPath()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc308 std::vector<const Edge*> output_edges; in RemoveIdentityNodesForArgRetval() local
310 output_edges.push_back(e); in RemoveIdentityNodesForArgRetval()
312 for (const Edge* e : output_edges) { in RemoveIdentityNodesForArgRetval()
1436 std::vector<std::vector<const Edge*>> output_edges(num_replicas * in RemoveUnusedXlaOutput() local
1441 output_edges[e->src_output()].push_back(e); in RemoveUnusedXlaOutput()
1448 for (const Edge* e : output_edges[i * num_replicated_outputs + j]) { in RemoveUnusedXlaOutput()
1458 TF_RET_CHECK(output_edges[i * num_replicated_outputs + j].empty()) in RemoveUnusedXlaOutput()
1460 << output_edges[i * num_replicated_outputs + j][0]->DebugString(); in RemoveUnusedXlaOutput()