Searched refs:kXlaClusterAttr (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 138 clustered_producer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
139 clustered_consumer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
175 clustered_producer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
176 clustered_consumer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
177 consumer_in_different_cluster->AddAttr(kXlaClusterAttr, "cluster_1"); in TEST()
205 clustered_producer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
206 clustered_consumer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
207 consumer_in_different_cluster->AddAttr(kXlaClusterAttr, "cluster_1"); in TEST()
236 clustered_producer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
237 clustered_consumer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
[all …]
|
| D | mark_for_compilation_pass.h | 31 extern const char* const kXlaClusterAttr;
|
| D | xla_cluster_util.cc | 39 const char* const kXlaClusterAttr = "_XlaCluster"; variable
198 const AttrValue* attr_value = node.attrs().Find(kXlaClusterAttr); in GetXlaClusterForNode()
217 node_def->mutable_attr()->erase(kXlaClusterAttr); in RemoveFromXlaCluster()
220 void RemoveFromXlaCluster(Node* node) { node->ClearAttr(kXlaClusterAttr); } in RemoveFromXlaCluster()
|
| D | xla_cluster_util.h | 35 extern const char* const kXlaClusterAttr;
|
| D | mark_for_compilation_pass.cc | 886 n->AddAttr(kXlaClusterAttr, name); in CreateClusters()
|
| D | encapsulate_subgraphs_pass.cc | 1306 kXlaClusterAttr, **options.graph, rewrite_subgraph, in Run()
|
| D | mark_for_compilation_pass_test.cc | 59 if (TryGetNodeAttr(node->attrs(), kXlaClusterAttr, &cluster)) { in GetClusters()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_functional_ops.cc | 39 constexpr const char* const kXlaClusterAttr = "_xla_compile_id"; variable
68 return CheckStringAttr(n, kXlaClusterAttr); in MarkedForXlaCompilation()
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | auto_clustering_test_helper.cc | 86 const char* kXlaClusterAttr = "_XlaCluster"; in AssertGraphDefIsUnclustered() local
90 if (node.attr().count(kXlaClusterAttr) || in AssertGraphDefIsUnclustered()
|