Searched refs:kXlaInternalScopeAttr (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | cluster_scoping_pass_test.cc | 52 if (GetNodeAttr(node->attrs(), kXlaInternalScopeAttr, &scope).ok()) { in GetXlaInternalScopes() 155 kXlaInternalScopeAttr, "ClusterA")); in TEST() 158 kXlaInternalScopeAttr, "ClusterA")); in TEST() 161 kXlaInternalScopeAttr, "ClusterB")); in TEST() 164 kXlaInternalScopeAttr, "ClusterD")); in TEST()
|
D | cluster_scoping_pass.cc | 56 if (GetNodeAttr(node->attrs(), kXlaInternalScopeAttr, &scope).ok()) { in GetXlaInternalScope() 64 node->AddAttr(kXlaInternalScopeAttr, scope); in SetXlaInternalScope()
|
D | defs.cc | 29 const char* const kXlaInternalScopeAttr = "_XlaInternalScope"; variable
|
D | defs.h | 36 extern const char* const kXlaInternalScopeAttr; // "_XlaInternalScope"
|
D | compilability_check_util.cc | 726 HasBoolAttr(node, kXlaInternalScopeAttr) || in NodeCanTriggerXlaCompilation()
|
D | mark_for_compilation_pass.cc | 960 GetNodeAttrString(node->attrs(), kXlaInternalScopeAttr); in GetXlaScope()
|