Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc903 shape.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
910 shape.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
960 b2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
970 kXlaHasHostTransferAttrName, true)); in TEST()
1037 shape1.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1044 shape1.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1054 shape2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1062 shape2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1074 shape2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1148 b2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
[all …]
Dextract_outside_compilation_pass.cc46 if (HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in HostGraphControlRetMapping()
122 recv_at_host_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildRecvAtHostNode()
224 send_from_host_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildSendFromHostNode()
401 if (!HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in ResetDeviceOrdinalToPlaceholderValue()
432 kXlaHasHostTransferAttrName, ": ", in ResetDeviceOrdinalToPlaceholderValue()
942 if (!HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in PostprocessLiftedArgs()
1075 if (HasNodeAttr(copy->def(), kXlaHasHostTransferAttrName)) { in ConstructHostGraph()
1417 recv_pred_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForIfNode()
1445 if_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForIfNode()
1592 recv_pred_builder.Attr(kXlaHasHostTransferAttrName, true); in RewriteHostWhileLoopCond()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc27 const char kXlaHasHostTransferAttrName[] = "_xla_has_host_transfer"; variable
38 if (!HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)) { in SetDeviceOrdinalAttributeForNode()
41 kXlaHasHostTransferAttrName); in SetDeviceOrdinalAttributeForNode()
Dside_effect_util.h39 extern const char kXlaHasHostTransferAttrName[];
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc300 TF_RET_CHECK(HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)); in SetNodeDeviceForTPUCommunication()
3479 if (HasNodeAttr(image->def(), kXlaHasHostTransferAttrName)) { in CopyOutsideCompilationNodes()
3540 if (HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)) { in ReplicateOutsideCompilationNodes()
3770 HasNodeAttr(n->def(), kXlaHasHostTransferAttrName); in LowerOutsideCompilationFunctionalNodes()
3808 node->AddAttr(kXlaHasHostTransferAttrName, true); in LowerOutsideCompilationFunctionalNodes()