Searched refs:kOutsideCompilationAttr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_defs.cc | 28 const char* const kOutsideCompilationAttr = "_xla_outside_compilation"; variable
|
| D | tpu_defs.h | 51 extern const char* const kOutsideCompilationAttr;
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 691 !TryGetNodeAttr(n->def(), kOutsideCompilationAttr, in MoveHeadOutsideCompilationToHost()
703 ph_builder.Attr(kOutsideCompilationAttr, outside_compilation_attr); in MoveHeadOutsideCompilationToHost()
1269 !HasNodeAttr(e->dst()->def(), kOutsideCompilationAttr)) { in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1338 TF_RETURN_IF_ERROR(GetNodeAttr(e->dst()->def(), kOutsideCompilationAttr, in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1340 ph_builder.Attr(kOutsideCompilationAttr, outside_compilation_attr); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1730 HasNodeAttr(e->dst()->def(), kOutsideCompilationAttr)) in FindArgsToLiftForIfNode()
1741 HasNodeAttr(e->dst()->def(), kOutsideCompilationAttr)) in FindArgsToLiftForIfNode()
1833 !HasNodeAttr(e->dst()->def(), kOutsideCompilationAttr)) { in FindArgsToLiftForWhileNode()
1867 if (!HasNodeAttr(e->dst()->def(), kOutsideCompilationAttr)) { in FindArgsToLiftForCallNode()
1923 id_builder.Attr(kOutsideCompilationAttr, id_builder.node_name()); in MakeIdentityNodesForArgsToLift()
[all …]
|
| D | distributed_tpu_rewrite_pass.cc | 352 node->attrs().Find(kOutsideCompilationAttr); in FindTaggedNodes()
356 kOutsideCompilationAttr, " attr but no ", in FindTaggedNodes()
374 kOutsideCompilationAttr, " attr."); in FindTaggedNodes()
378 return errors::Internal("Attr ", kOutsideCompilationAttr, " on node ", in FindTaggedNodes()
3599 GetNodeAttr(src->def(), kOutsideCompilationAttr, in CopyOutsideCompilationEdges()
3718 if (!HasNodeAttr(n->def(), kOutsideCompilationAttr)) { in LowerOutsideCompilationFunctionalNodes()
3752 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), kOutsideCompilationAttr, in LowerOutsideCompilationFunctionalNodes()
3815 node->AddAttr(kOutsideCompilationAttr, outside_compilation_attr); in LowerOutsideCompilationFunctionalNodes()
3863 n->ClearAttr(kOutsideCompilationAttr); in LowerOutsideCompilationFunctionalNodes()
4203 node->ClearAttr(kOutsideCompilationAttr); in UpdateHeadTailOutsideCompilation()
|