Searched refs:cond_host_func_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
1557 xla_cluster_name, cond_host_func_name, fld)); in RewriteHostWhileLoopCond()
1565 const FunctionDef* cond_host_func = fld->Find(cond_host_func_name); in RewriteHostWhileLoopCond()
1578 cond_host_func_name); in RewriteHostWhileLoopCond()
1616 TF_RETURN_IF_ERROR(GraphToFunctionDef(*cond_graph, cond_host_func_name, in RewriteHostWhileLoopCond()
1620 fld->ReplaceFunction(cond_host_func_name, cond_replace_fdef)); in RewriteHostWhileLoopCond()
1691 const string& cond_host_func_name, const string& body_host_func_name) { in BuildHostGraphForWhileNode() argument
1702 cond_host_func_name, while_node_name, host_transfer_key, in BuildHostGraphForWhileNode()
1720 func.set_name(cond_host_func_name); in BuildHostGraphForWhileNode()
2026 string cond_host_func_name = absl::StrCat("oc_cond_host_while_", cond.name()), in ExtractOutsideCompilationForWhileNode() local
[all …]
|