Searched refs:kXlaClusterAttr (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 141 clustered_producer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
142 clustered_consumer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
178 clustered_producer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
179 clustered_consumer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
180 consumer_in_different_cluster->AddAttr(kXlaClusterAttr, "cluster_1"); in TEST()
208 clustered_producer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
209 clustered_consumer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
210 consumer_in_different_cluster->AddAttr(kXlaClusterAttr, "cluster_1"); in TEST()
239 clustered_producer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
240 clustered_consumer->AddAttr(kXlaClusterAttr, "cluster_0"); in TEST()
[all …]
|
| D | encapsulate_xla_computations_pass.cc | 36 const char* const EncapsulateXlaComputationsPass::kXlaClusterAttr = member in tensorflow::EncapsulateXlaComputationsPass
48 EncapsulateXlaComputationsPass::kXlaClusterAttr, &name) in IsCpuGpuCompile()
182 AddNodeAttr(EncapsulateXlaComputationsPass::kXlaClusterAttr, call_def->name(), in RewriteSubgraph()
218 e->src()->attrs().Find(kXlaClusterAttr) != nullptr && in Encapsulate()
219 e->dst()->attrs().Find(kXlaClusterAttr) == nullptr && in Encapsulate()
235 kXlaClusterAttr, "", **graph, RewriteSubgraph, in Encapsulate()
249 if (GetNodeAttr(n->attrs(), kXlaClusterAttr, &name).ok()) { in BuildXlaLaunchOps()
|
| D | xla_cluster_util.cc | 37 const char* const kXlaClusterAttr = "_XlaCluster"; variable
205 const AttrValue* attr_value = node.attrs().Find(kXlaClusterAttr); in GetXlaClusterForNode()
224 node_def->mutable_attr()->erase(kXlaClusterAttr); in RemoveFromXlaCluster()
227 void RemoveFromXlaCluster(Node* node) { node->ClearAttr(kXlaClusterAttr); } in RemoveFromXlaCluster()
|
| D | xla_fusion_optimizer.cc | 171 status = GetNodeAttr(node->attrs(), kXlaClusterAttr, &compile); in Optimize()
338 n->AddAttr(kXlaClusterAttr, name); in Optimize()
|
| D | encapsulate_xla_computations_pass.h | 37 static const char* const kXlaClusterAttr; // _xla_compile_id
|
| D | encapsulate_xla_computations_pass_test.cc | 59 .Attr(EncapsulateXlaComputationsPass::kXlaClusterAttr, "launch0") in MakeOuterGraph()
110 node->AddAttr(EncapsulateXlaComputationsPass::kXlaClusterAttr, "launch0"); in MakeBodyGraph()
158 node->AddAttr(EncapsulateXlaComputationsPass::kXlaClusterAttr, in TEST()
219 node->AddAttr(EncapsulateXlaComputationsPass::kXlaClusterAttr, "launch0"); in TEST()
|
| D | mark_for_compilation_pass.h | 29 extern const char* const kXlaClusterAttr;
|
| D | xla_cluster_util.h | 32 extern const char* const kXlaClusterAttr;
|
| D | xla_fusion_optimizer_test.cc | 38 if (GetNodeAttr(AttrSlice(node), kXlaClusterAttr, &cluster).ok()) { in GetClusters()
|
| D | mark_for_compilation_pass.cc | 1338 n->AddAttr(kXlaClusterAttr, name); in RunImpl()
|
| D | mark_for_compilation_pass_test.cc | 54 if (GetNodeAttr(node->attrs(), kXlaClusterAttr, &cluster).ok()) { in GetClusters()
|
| D | encapsulate_subgraphs_pass.cc | 2666 kXlaClusterAttr, kXlaOutsideCompilationAttr, **options.graph, in Run()
|