Home
last modified time | relevance | path

Searched refs:is_lifted_arg (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc922 bool is_lifted_arg; in OutsideCompilationAttrToNode() local
924 if (TryGetNodeAttr(n->def(), kXlaIsLiftedArgAttrName, &is_lifted_arg) && in OutsideCompilationAttrToNode()
927 TF_RET_CHECK(is_lifted_arg); in OutsideCompilationAttrToNode()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc688 bool is_lifted_arg; in MoveHeadOutsideCompilationToHost() local
690 if (!TryGetNodeAttr(n->def(), kXlaIsLiftedArgAttrName, &is_lifted_arg) || in MoveHeadOutsideCompilationToHost()
Ddistributed_tpu_rewrite_pass.cc3595 bool is_lifted_arg; in CopyOutsideCompilationEdges() local
3597 if (GetNodeAttr(src->def(), kXlaIsLiftedArgAttrName, &is_lifted_arg) in CopyOutsideCompilationEdges()