Searched refs:kXlaMustCompileAttr (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | force_xla_constants_on_host_pass_test.cc | 77 (*called_func.mutable_attr())[kXlaMustCompileAttr] = true_attribute; in TEST()
88 call.output.front().node()->AddAttr(kXlaMustCompileAttr, true); in TEST()
|
| D | defs.cc | 20 const char* const kXlaMustCompileAttr = "_XlaMustCompile"; variable
|
| D | defs.h | 28 extern const char* const kXlaMustCompileAttr; // "_XlaMustCompile"
|
| D | compilability_check_util_test.cc | 494 (*identity_func.mutable_attr())[kXlaMustCompileAttr] = true_attribute; in TEST_F()
508 {kXlaMustCompileAttr, true}}}}, in TEST_F()
|
| D | compilability_check_util.cc | 618 return HasBoolAttr(node_def, kXlaMustCompileAttr); in CanCreateXlaKernel()
723 HasBoolAttr(node, kXlaMustCompileAttr) || in NodeCanTriggerXlaCompilation()
|
| D | mark_for_compilation_pass.cc | 1047 GetNodeOrFuncAttr(node, flib_def_, kXlaMustCompileAttr)); in BuildInitialClusterSet()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.cc | 104 const auto& it = ndef.attr().find(kXlaMustCompileAttr); in MemoryTypesForNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 363 Status status = GetFuncAttr(op, ctx, kXlaMustCompileAttr, compile_with_xla); in MustCompileWithXLA()
479 op->MutableAttrs()->Set(kXlaMustCompileAttr, true); in GetOrCreateKernelAndDevice()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 823 is_enabled(kXlaMustCompileAttr); in MarkedForXlaCompilation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 584 TryGetNodeAttr(AttrSlice(&fbody->fdef.attr()), kXlaMustCompileAttr, in GetGraph()
|