Searched refs:host_transfer_key (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1325 const string& name, const string& host_transfer_key, Node* pred_node, in BuildSendIfPredNode() argument
1329 send_pred_builder.Attr("key", absl::StrCat(host_transfer_key, "_dtoh_0")); in BuildSendIfPredNode()
1394 const string& if_node_name, const string& host_transfer_key, in BuildHostGraphForIfNode() argument
1412 recv_pred_builder.Attr("key", host_transfer_key); in BuildHostGraphForIfNode()
1475 const string& cond_xla_func_name, const string& host_transfer_key, in AddSendLoopPredToLoopCond() argument
1512 absl::StrCat(host_transfer_key, "_dtoh_0")); in AddSendLoopPredToLoopCond()
1552 const string& host_transfer_key, const string& xla_cluster_attr_name, in RewriteHostWhileLoopCond() argument
1585 recv_pred_builder.Attr("key", host_transfer_key); in RewriteHostWhileLoopCond()
1628 const string& host_transfer_key, const string& xla_cluster_attr_name, in RewriteHostWhileLoopBody() argument
1689 const string& while_node_name, const string& host_transfer_key, in BuildHostGraphForWhileNode() argument
[all …]
|