Searched refs:oc_output_edges (Results 1 – 1 of 1) sorted by relevance
518 std::vector<const Edge*> oc_output_edges; in MoveHeadOutsideCompilationToHost() local525 oc_output_edges.push_back(e); in MoveHeadOutsideCompilationToHost()531 old_num_per_replica_inputs + oc_output_edges.size(); in MoveHeadOutsideCompilationToHost()572 /*additional_per_replica_inputs=*/oc_output_edges.size(), xla_node)); in MoveHeadOutsideCompilationToHost()636 oc_output_edges[input_index - old_num_per_replica_inputs]->src(); in MoveHeadOutsideCompilationToHost()638 oc_output_edges[input_index - old_num_per_replica_inputs] in MoveHeadOutsideCompilationToHost()673 const Edge* original_edge = oc_output_edges[i - old_num_per_replica_inputs]; in MoveHeadOutsideCompilationToHost()