Searched refs:kXlaScopeAttr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | defs.cc | 25 const char* const kXlaScopeAttr = "_XlaScope"; variable
|
| D | defs.h | 35 extern const char* const kXlaScopeAttr; // "_XlaScope"
|
| D | mark_for_compilation_pass_test.cc | 454 .WithAttr(kXlaScopeAttr, "ScopeA")); in TEST()
457 builder.opts().WithName("B").WithAttr(kXlaScopeAttr, "ScopeB")); in TEST()
460 builder.opts().WithName("C").WithAttr(kXlaScopeAttr, "ScopeC")); in TEST()
487 .WithAttr(kXlaScopeAttr, "ScopeA")); in TEST()
490 builder.opts().WithName("B").WithAttr(kXlaScopeAttr, "ScopeB")); in TEST()
493 builder.opts().WithName("C").WithAttr(kXlaScopeAttr, "ScopeC")); in TEST()
516 .WithAttr(kXlaScopeAttr, "Scope1")); in TEST()
521 .WithAttr(kXlaScopeAttr, "Scope1")); in TEST()
526 .WithAttr(kXlaScopeAttr, "Scope2")); in TEST()
531 .WithAttr(kXlaScopeAttr, "Scope2")); in TEST()
[all …]
|
| D | compilability_check_util.cc | 725 HasBoolAttr(node, kXlaScopeAttr) || in NodeCanTriggerXlaCompilation()
|
| D | mark_for_compilation_pass.cc | 966 const string& scope = GetNodeAttrString(node->attrs(), kXlaScopeAttr); in GetXlaScope()
|