Searched refs:GetXlaClusterForNode (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 50 absl::optional<absl::string_view> from_cluster = GetXlaClusterForNode(*n); in FindNodesToDecluster()
109 result->count(dst) ? absl::nullopt : GetXlaClusterForNode(*dst); in FindNodesToDecluster()
120 absl::string_view cluster_name = *GetXlaClusterForNode(*n); in PartiallyDeclusterNode()
129 GetXlaClusterForNode(*dst); in PartiallyDeclusterNode()
233 GetXlaClusterForNode(*edge.src()); in IsIntraClusterEdge()
235 GetXlaClusterForNode(*edge.dst()); in IsIntraClusterEdge()
314 absl::string_view cluster_name = *GetXlaClusterForNode(*n); in PartiallyDeclusterGraph()
318 GetXlaClusterForNode(*e->src()); in PartiallyDeclusterGraph()
369 absl::optional<absl::string_view> cluster = GetXlaClusterForNode(*n); in PartiallyDeclusterGraph()
375 return cluster == GetXlaClusterForNode(*e->src()); in PartiallyDeclusterGraph()
|
| D | partially_decluster_pass_test.cc | 319 EXPECT_EQ(GetXlaClusterForNode(*n), absl::nullopt); in TEST()
347 EXPECT_EQ(GetXlaClusterForNode(*n), "cluster_0"); in TEST()
372 EXPECT_EQ(GetXlaClusterForNode(*n), "cluster_1"); in TEST()
406 EXPECT_EQ(GetXlaClusterForNode(*n), "cluster_0"); in TEST()
453 EXPECT_EQ(GetXlaClusterForNode(*n_b), absl::nullopt); in TEST()
457 EXPECT_EQ(GetXlaClusterForNode(*n_c), absl::nullopt); in TEST()
461 EXPECT_EQ(GetXlaClusterForNode(*n_d), absl::nullopt); in TEST()
465 EXPECT_EQ(GetXlaClusterForNode(*n_e), absl::nullopt); in TEST()
485 EXPECT_EQ(GetXlaClusterForNode(*n_b), "cluster_0"); in TEST()
489 EXPECT_EQ(GetXlaClusterForNode(*n_c), "cluster_0"); in TEST()
|
| D | xla_cluster_util.h | 66 absl::optional<absl::string_view> GetXlaClusterForNode(const Node& node);
|
| D | increase_dynamism_for_auto_jit_pass.cc | 323 if (!GetXlaClusterForNode(*n).has_value()) { in ShouldRewriteSlice()
360 RewriteSlice(g, n, *slice_inputs, *GetXlaClusterForNode(*n))); in FindAndRewriteSlices()
|
| D | xla_cluster_util.cc | 197 absl::optional<absl::string_view> GetXlaClusterForNode(const Node& node) { in GetXlaClusterForNode() function
361 absl::optional<absl::string_view> cluster_name = GetXlaClusterForNode(*n); in GetXlaAutoClusteringSummary()
|
| D | mark_for_compilation_pass.cc | 1455 GetXlaClusterForNode(*n)) { in DumpPostClusteringGraphs()
1541 GetXlaClusterForNode(*from); in VLogClusteringSummary()
1545 GetXlaClusterForNode(*to); in VLogClusteringSummary()
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | auto_clustering_test_helper.cc | 54 GetXlaClusterForNode(*n)) { in SummarizeClustering()
|