Searched refs:kXlaCompileAttr (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op.cc | 84 const auto& it = node_def.attr().find(kXlaCompileAttr); in CompilationRequested()
93 node_def, kXlaCompileAttr, &xla_compile); in CompilationRequested()
97 VLOG(3) << "No " << kXlaCompileAttr << " attr defined for " in CompilationRequested()
|
| D | mark_for_compilation_pass.cc | 724 Status status = GetNodeAttr(node->attrs(), kXlaCompileAttr, &compile); in Run()
728 << kXlaCompileAttr << ") is false."; in Run()
733 status = fld->GetAttr(*node, kXlaCompileAttr, &compile); in Run()
737 << kXlaCompileAttr << ") on callee is false."; in Run()
1124 if (GetNodeAttr(node->attrs(), kXlaCompileAttr, &xla_compile_attr).ok()) { in RunImpl()
1127 if (options.flib_def->GetAttr(*node, kXlaCompileAttr, &xla_compile_attr) in RunImpl()
1319 if (GetNodeAttr(n->attrs(), kXlaCompileAttr, &compile_attr).ok()) { in RunImpl()
1321 } else if (options.flib_def->GetAttr(*n, kXlaCompileAttr, &compile_attr) in RunImpl()
|
| D | defs.cc | 20 const char* const kXlaCompileAttr = "_XlaCompile"; variable
|
| D | defs.h | 25 extern const char* const kXlaCompileAttr; // "_XlaCompile"
|
| D | mark_for_compilation_pass_test.cc | 452 .WithAttr(kXlaCompileAttr, true) in TEST()
457 .WithAttr(kXlaCompileAttr, true) in TEST()
462 .WithAttr(kXlaCompileAttr, true) in TEST()
467 .WithAttr(kXlaCompileAttr, true) in TEST()
495 .WithAttr(kXlaCompileAttr, true) in TEST()
500 .WithAttr(kXlaCompileAttr, true) in TEST()
696 c.node()->AddAttr(kXlaCompileAttr, true); in TEST()
707 c.node()->AddAttr(kXlaCompileAttr, true); in TEST()
|
| D | xla_fusion_optimizer.cc | 164 Status status = GetNodeAttr(node->attrs(), kXlaCompileAttr, &compile); in Optimize()
|
| D | xla_fusion_optimizer_test.cc | 122 builder.opts().WithName("F").WithAttr(kXlaCompileAttr, true)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 54 const char* const kXlaCompileAttr = "_XlaCompile"; variable
401 op->MutableAttrs()->Set(kXlaCompileAttr, true); in EagerLocalExecute()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 220 AddNodeAttr(kXlaCompileAttr, true, test_def); in BuildGraph()
|