Searched refs:replicate_nodes (Results 1 – 2 of 2) sorted by relevance
317 Graph* graph, std::vector<Node*>* replicate_nodes, in FindTaggedNodes() argument323 replicate_nodes->push_back(node); in FindTaggedNodes()500 const Graph* graph, const std::vector<Node*>& replicate_nodes, in FindHostComputeKeyPlaceholderNodes() argument503 for (const auto node : replicate_nodes) { in FindHostComputeKeyPlaceholderNodes()4449 std::vector<Node*> replicate_nodes; in Run() local4454 TF_RETURN_IF_ERROR(FindTaggedNodes(graph, &replicate_nodes, in Run()4458 if (replicate_nodes.empty()) { in Run()4468 graph, replicate_nodes, &host_compute_key_placeholder_map)); in Run()4479 for (Node* node : replicate_nodes) { in Run()
2545 std::vector<Node*> replicate_nodes; in BuildTPUReplicateOps() local2551 replicate_nodes.push_back(n); in BuildTPUReplicateOps()2600 for (Node* replicate : replicate_nodes) { in BuildTPUReplicateOps()