Searched refs:kXlaHasHostTransferAttrName (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 910 shape.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
917 shape.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
962 b2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
971 kXlaHasHostTransferAttrName, true)); in TEST()
1038 shape1.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1045 shape1.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1055 shape2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1063 shape2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1075 shape2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
1138 b2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); in TEST()
[all …]
|
| D | extract_outside_compilation_pass.cc | 109 recv_at_host_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildRecvAtHostNode()
211 send_from_host_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildSendFromHostNode()
412 if (!HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in ResetDeviceOrdinalToPlaceholderValue()
443 kXlaHasHostTransferAttrName, ": ", in ResetDeviceOrdinalToPlaceholderValue()
554 if (HasNodeAttr(copy->def(), kXlaHasHostTransferAttrName)) { in ConstructHostGraph()
902 recv_pred_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForIfNode()
930 if_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForIfNode()
1201 while_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForWhileNode()
1254 call_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForFuncCallNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 27 const char kXlaHasHostTransferAttrName[] = "_xla_has_host_transfer"; variable
30 if (!HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)) { in SetDeviceOrdinalAttributeForNode()
33 kXlaHasHostTransferAttrName); in SetDeviceOrdinalAttributeForNode()
|
| D | side_effect_util.h | 39 extern const char kXlaHasHostTransferAttrName[];
|