Searched refs:original_edge (Results 1 – 1 of 1) sorted by relevance
490 const Edge* original_edge = input_edges.at(new_index); in MoveHeadOutsideCompilationToHost() local491 g->AddEdge(original_edge->src(), original_edge->src_output(), in MoveHeadOutsideCompilationToHost()673 const Edge* original_edge = oc_output_edges[i - old_num_per_replica_inputs]; in MoveHeadOutsideCompilationToHost() local674 Node* dst = original_edge->dst(); in MoveHeadOutsideCompilationToHost()675 int dst_input = original_edge->dst_input(); in MoveHeadOutsideCompilationToHost()676 xla_graph->RemoveEdge(original_edge); in MoveHeadOutsideCompilationToHost()1140 const Edge* original_edge; in MoveTailOutsideCompilationToHost() local1141 Status s = output.first->input_edge(output.second, &original_edge); in MoveTailOutsideCompilationToHost()1143 g->RemoveEdge(original_edge); in MoveTailOutsideCompilationToHost()1173 const Edge* original_edge = oc_input_edges[i]; in MoveTailOutsideCompilationToHost() local[all …]