Home
last modified time | relevance | path

Searched refs:GetXlaClusterForNode (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass.cc47 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()
Dpartially_decluster_pass_test.cc322 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()
Dxla_cluster_util.h66 absl::optional<absl::string_view> GetXlaClusterForNode(const Node& node);
Dincrease_dynamism_for_auto_jit_pass.cc323 if (!GetXlaClusterForNode(*n).has_value()) { in ShouldRewriteSlice()
360 RewriteSlice(g, n, *slice_inputs, *GetXlaClusterForNode(*n))); in FindAndRewriteSlices()
Dmark_for_compilation_pass.cc784 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()
Dxla_cluster_util.cc204 absl::optional<absl::string_view> GetXlaClusterForNode(const Node& node) { in GetXlaClusterForNode() function