| 292 | } |
| 293 | |
| 294 | void AddToCluster(absl::Span<Node* const> nodes, |
| 295 | absl::string_view cluster_name) { |
| 296 | for (Node* n : nodes) { |
| 297 | n->AddAttr(kXlaClusterAttr, string(cluster_name)); |
| 298 | } |
| 299 | } |
| 300 | |
| 301 | TEST(PartiallyDeclusterPassTest, DeclusterMustBeConstantNodes) { |
| 302 | tensorflow::Scope s = tensorflow::Scope::NewRootScope(); |