Home
last modified time | relevance | path

Searched refs:kXlaIsLiftedArgAttrName (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.h66 extern const char kXlaIsLiftedArgAttrName[];
Dencapsulate_util.cc318 const char kXlaIsLiftedArgAttrName[] = "_xla_is_lifted_arg"; variable
Dextract_outside_compilation_pass.cc924 if (TryGetNodeAttr(n->def(), kXlaIsLiftedArgAttrName, &is_lifted_arg) && in OutsideCompilationAttrToNode()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc690 if (!TryGetNodeAttr(n->def(), kXlaIsLiftedArgAttrName, &is_lifted_arg) || in MoveHeadOutsideCompilationToHost()
702 ph_builder.Attr(kXlaIsLiftedArgAttrName, true); in MoveHeadOutsideCompilationToHost()
1924 id_builder.Attr(kXlaIsLiftedArgAttrName, true); in MakeIdentityNodesForArgsToLift()
Ddistributed_tpu_rewrite_pass.cc3597 if (GetNodeAttr(src->def(), kXlaIsLiftedArgAttrName, &is_lifted_arg) in CopyOutsideCompilationEdges()