Searched refs:while_node_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1551 const string& cond_host_func_name, const string& while_node_name, in RewriteHostWhileLoopCond() argument
1583 absl::StrCat("recv_oc_while_cond_", while_node_name), "_XlaRecvAtHost"); in RewriteHostWhileLoopCond()
1601 absl::StrCat("recv_oc_while_cond_ret_", while_node_name), "_Retval"); in RewriteHostWhileLoopCond()
1627 const string& body_host_func_name, const string& while_node_name, in RewriteHostWhileLoopBody() argument
1660 absl::StrCat("recv_oc_while_body_ret_", while_node_name), "_Retval"); in RewriteHostWhileLoopBody()
1689 const string& while_node_name, const string& host_transfer_key, in BuildHostGraphForWhileNode() argument
1693 string outside_compilation_name = absl::StrCat("oc_while_", while_node_name); in BuildHostGraphForWhileNode()
1702 cond_host_func_name, while_node_name, host_transfer_key, in BuildHostGraphForWhileNode()
1708 body_host_func_name, while_node_name, host_transfer_key, in BuildHostGraphForWhileNode()
1713 NodeDefBuilder while_builder(absl::StrCat("oc_while_", while_node_name), in BuildHostGraphForWhileNode()
|