Home
last modified time | relevance | path

Searched refs:output_edge (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc95 for (const Edge* output_edge : enter_node->out_edges()) { in ResourceOnlyUsedForTPUExecuteInLoop() local
96 Node* output_node = output_edge->dst(); in ResourceOnlyUsedForTPUExecuteInLoop()
97 if (output_edge->IsControlEdge() || output_node->IsExit()) continue; in ResourceOnlyUsedForTPUExecuteInLoop()
Dencapsulate_tpu_computations_pass.cc1536 for (const Edge* output_edge : xla_node_out_edges[output_index]) { in RemoveEdgesBetweenArgAndRetval() local
1537 Node* dst = output_edge->dst(); in RemoveEdgesBetweenArgAndRetval()
1538 int dst_input = output_edge->dst_input(); in RemoveEdgesBetweenArgAndRetval()
1540 g->RemoveEdge(output_edge); in RemoveEdgesBetweenArgAndRetval()