Searched refs:body_host_func_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1627 const string& body_host_func_name, const string& while_node_name, in RewriteHostWhileLoopBody() argument
1633 xla_cluster_name, body_host_func_name, fld)); in RewriteHostWhileLoopBody()
1641 const FunctionDef* body_host_func = fld->Find(body_host_func_name); in RewriteHostWhileLoopBody()
1655 body_host_func_name); in RewriteHostWhileLoopBody()
1676 TF_RETURN_IF_ERROR(GraphToFunctionDef(*body_graph, body_host_func_name, in RewriteHostWhileLoopBody()
1680 fld->ReplaceFunction(body_host_func_name, body_replace_fdef)); in RewriteHostWhileLoopBody()
1691 const string& cond_host_func_name, const string& body_host_func_name) { in BuildHostGraphForWhileNode() argument
1708 body_host_func_name, while_node_name, host_transfer_key, in BuildHostGraphForWhileNode()
1722 func.set_name(body_host_func_name); in BuildHostGraphForWhileNode()
2027 body_host_func_name = absl::StrCat("oc_body_host_while_", body.name()); in ExtractOutsideCompilationForWhileNode() local
[all …]
|