Searched refs:kXlaCompileAttr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | defs.cc | 22 const char* const kXlaCompileAttr = "_XlaCompile"; variable
|
| D | defs.h | 32 extern const char* const kXlaCompileAttr; // "_XlaCompile"
|
| D | mark_for_compilation_pass.cc | 1045 GetNodeOrFuncAttr(node, flib_def_, kXlaCompileAttr) || in BuildInitialClusterSet()
1324 Status status = GetNodeAttr(node->attrs(), kXlaCompileAttr, &compile); in CompilationDisallowedByXlaCompileAttr()
1328 << kXlaCompileAttr << ") is false."; in CompilationDisallowedByXlaCompileAttr()
1333 status = flib_def_->GetAttr(*node, kXlaCompileAttr, &compile); in CompilationDisallowedByXlaCompileAttr()
1337 << kXlaCompileAttr << ") on callee is false."; in CompilationDisallowedByXlaCompileAttr()
|
| D | mark_for_compilation_pass_test.cc | 515 .WithAttr(kXlaCompileAttr, true) in TEST()
520 .WithAttr(kXlaCompileAttr, true) in TEST()
525 .WithAttr(kXlaCompileAttr, true) in TEST()
530 .WithAttr(kXlaCompileAttr, true) in TEST()
557 .WithAttr(kXlaCompileAttr, true) in TEST()
562 .WithAttr(kXlaCompileAttr, true) in TEST()
828 c.node()->AddAttr(kXlaCompileAttr, true); in TEST()
839 c.node()->AddAttr(kXlaCompileAttr, true); in TEST()
|
| D | compilability_check_util.cc | 724 HasBoolAttr(node, kXlaCompileAttr) || in NodeCanTriggerXlaCompilation()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 221 AddNodeAttr(kXlaCompileAttr, true, test_def); in BuildGraph()
|