Searched refs:GetXlaClusterForNode (Results 1 – 6 of 6) sorted by relevance
47 absl::optional<absl::string_view> from_cluster = GetXlaClusterForNode(*n); in FindNodesToDecluster()97 result->count(dst) ? absl::nullopt : GetXlaClusterForNode(*dst); in FindNodesToDecluster()108 absl::string_view cluster_name = *GetXlaClusterForNode(*n); in PartiallyDeclusterNode()117 GetXlaClusterForNode(*dst); in PartiallyDeclusterNode()221 GetXlaClusterForNode(*edge.src()); in IsIntraClusterEdge()223 GetXlaClusterForNode(*edge.dst()); in IsIntraClusterEdge()302 absl::string_view cluster_name = *GetXlaClusterForNode(*n); in PartiallyDeclusterGraph()306 GetXlaClusterForNode(*e->src()); in PartiallyDeclusterGraph()
322 EXPECT_EQ(GetXlaClusterForNode(*n), absl::nullopt); in TEST()350 EXPECT_EQ(GetXlaClusterForNode(*n), "cluster_0"); in TEST()375 EXPECT_EQ(GetXlaClusterForNode(*n), "cluster_1"); in TEST()409 EXPECT_EQ(GetXlaClusterForNode(*n), "cluster_0"); in TEST()440 EXPECT_EQ(GetXlaClusterForNode(*n), "cluster_0"); in TEST()
66 absl::optional<absl::string_view> GetXlaClusterForNode(const Node& node);
323 if (!GetXlaClusterForNode(*n).has_value()) { in ShouldRewriteSlice()360 RewriteSlice(g, n, *slice_inputs, *GetXlaClusterForNode(*n))); in FindAndRewriteSlices()
784 absl::optional<absl::string_view> cluster_name = GetXlaClusterForNode(*n); in VLogClusteringSummary()848 GetXlaClusterForNode(*from); in VLogClusteringSummary()852 GetXlaClusterForNode(*to); in VLogClusteringSummary()1355 GetXlaClusterForNode(*n)) { in RunImpl()
204 absl::optional<absl::string_view> GetXlaClusterForNode(const Node& node) { in GetXlaClusterForNode() function