Searched defs:host_transfer_key (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass.cc | 1340 const string& name, const string& host_transfer_key, Node* pred_node, in BuildSendIfPredNode()
1408 const string& if_node_name, const string& host_transfer_key, in BuildHostGraphForIfNode()
1486 const string& cond_xla_func_name, const string& host_transfer_key, in AddSendLoopPredToLoopCond()
1563 const string& host_transfer_key, const string& xla_cluster_attr_name, in RewriteHostWhileLoopCond()
1637 const string& host_transfer_key, const string& xla_cluster_attr_name, in RewriteHostWhileLoopBody()
1696 const string& while_node_name, const string& host_transfer_key, in BuildHostGraphForWhileNode()
1948 string host_transfer_key = absl::StrCat("oc_if_pred_", n->name()); in ExtractOutsideCompilationForIfNode() local
2062 string host_transfer_key = absl::StrCat("oc_while_pred_", n->name()); in ExtractOutsideCompilationForWhileNode() local
|