Searched refs:replica_out_edges (Results 1 – 1 of 1) sorted by relevance
2795 std::vector<std::vector<const Edge*>> replica_out_edges(num_replicas); in BuildTPUReplicateOps() local2799 replica_out_edges[e->src_output()].push_back(e); in BuildTPUReplicateOps()2804 for (const Edge* edge : replica_out_edges[replica]) { in BuildTPUReplicateOps()