Searched refs:host_graph_func_name (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1121 const string& host_graph_func_name, in ExpandHostGraphIntoMainGraph() argument
1133 const FunctionDef* host_graph_func = fld->Find(host_graph_func_name); in ExpandHostGraphIntoMainGraph()
1395 const string& host_graph_func_name, FunctionLibraryDefinition* fld, in BuildHostGraphForIfNode() argument
1461 TF_RETURN_IF_ERROR(GraphToFunctionDef(host_graph, host_graph_func_name, in BuildHostGraphForIfNode()
1463 if (fld->Find(host_graph_func_name)) { in BuildHostGraphForIfNode()
1465 fld->ReplaceFunction(host_graph_func_name, oc_host_graph_fdef)); in BuildHostGraphForIfNode()
1690 const string& host_graph_func_name, FunctionLibraryDefinition* fld, in BuildHostGraphForWhileNode() argument
1742 TF_RETURN_IF_ERROR(GraphToFunctionDef(host_graph, host_graph_func_name, in BuildHostGraphForWhileNode()
1744 if (fld->Find(host_graph_func_name)) { in BuildHostGraphForWhileNode()
1746 fld->ReplaceFunction(host_graph_func_name, oc_host_graph_fdef)); in BuildHostGraphForWhileNode()
[all …]
|
| D | extract_outside_compilation_pass.h | 93 const string& host_graph_func_name,
|
| D | extract_outside_compilation_pass_test.cc | 242 const string &new_func_name, const string &host_graph_func_name, in ExtractOutsideCompilationTest() argument
255 func_name_attrs, new_func_name, host_graph_func_name, host_compute_core, in ExtractOutsideCompilationTest()
|