Searched refs:cluster_nodes (Results 1 – 2 of 2) sorted by relevance
178 Status ReplicateNodes(const std::vector<Node*>& cluster_nodes, in ReplicateNodes() argument181 for (Node* n : cluster_nodes) { in ReplicateNodes()189 const std::vector<Node*>& cluster_nodes, Graph* graph) { in ReplicateEdges() argument190 for (const auto* node : cluster_nodes) { in ReplicateEdges()249 const std::vector<Node*>& cluster_nodes = it.second; in ReplicatePerReplicaNodesInFunctionGraph() local252 for (Node* n : cluster_nodes) { in ReplicatePerReplicaNodesInFunctionGraph()262 ReplicateNodes(cluster_nodes, allowed_devices, &helper, graph)); in ReplicatePerReplicaNodesInFunctionGraph()263 TF_RETURN_IF_ERROR(ReplicateEdges(helper, cluster_nodes, graph)); in ReplicatePerReplicaNodesInFunctionGraph()266 for (auto* n : cluster_nodes) { in ReplicatePerReplicaNodesInFunctionGraph()
51 const std::vector<const NodeDef*>& cluster_nodes, in FilterPartitionedConstNodes() argument53 for (const NodeDef* node : cluster_nodes) { in FilterPartitionedConstNodes()