Searched refs:tpu_replicate (Results 1 – 1 of 1) sorted by relevance
2841 Node* tpu_replicate = graph->AddNode(def, &status); in BuildTPUReplicateOps() local2846 graph->AddEdge(data_inputs[i].first, data_inputs[i].second, tpu_replicate, in BuildTPUReplicateOps()2850 graph->AddControlEdge(n, tpu_replicate); in BuildTPUReplicateOps()2854 graph->AddEdge(tpu_replicate, i, successor.first, successor.second); in BuildTPUReplicateOps()2858 graph->AddControlEdge(tpu_replicate, n); in BuildTPUReplicateOps()