Searched refs:new_func_name (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 96 const string& func_name, const string& new_func_name, in FunctionalizeControlFlowForFunction() argument
217 GraphToFunctionDef(*g, new_func_name, &functionalized_fdef)); in FunctionalizeControlFlowForFunction()
218 if (func_name == new_func_name) { in FunctionalizeControlFlowForFunction()
221 fld->ReplaceFunction(new_func_name, functionalized_fdef)); in FunctionalizeControlFlowForFunction()
223 VLOG(2) << "Adding function " << new_func_name; in FunctionalizeControlFlowForFunction()
266 string new_func_name = options.flib_def->UniqueFunctionName( in Run() local
270 func.name(), new_func_name, func.attr(), options.flib_def, flr, in Run()
274 func.set_name(new_func_name); in Run()
|
| D | tf2xla_util.cc | 203 string new_func_name = in PropagateConstIntoFuncAttr() local
206 GraphToFunctionDef(*func_graph, new_func_name, &replace_fdef)); in PropagateConstIntoFuncAttr()
210 func_attr.set_name(new_func_name); in PropagateConstIntoFuncAttr()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.h | 90 const NameAttrList& func_name_attrs, const string& new_func_name,
|
| D | extract_outside_compilation_pass.cc | 1310 string new_func_name = absl::StrCat(n->name(), "_oc"); in ExtractOutsideCompilationForNodesWithAssociatedFunctions() local
1314 func, new_func_name, host_func_name, host_compute_core, flr, fld, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1325 NodeDefBuilder replace_builder(n->name(), new_func_name, fld); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1587 const NameAttrList& func_name_attrs, const string& new_func_name, in ExtractOutsideCompilationForFunction() argument
1694 GraphToFunctionDef(*graph_out, new_func_name, &updated_fdef)); in ExtractOutsideCompilationForFunction()
1701 if (fld->Find(new_func_name)) { in ExtractOutsideCompilationForFunction()
1702 TF_RETURN_IF_ERROR(fld->ReplaceFunction(new_func_name, updated_fdef)); in ExtractOutsideCompilationForFunction()
|
| D | extract_outside_compilation_pass_test.cc | 242 const string &new_func_name, const string &host_graph_func_name, in ExtractOutsideCompilationTest() argument
254 func_name_attrs, new_func_name, host_graph_func_name, host_compute_core, in ExtractOutsideCompilationTest()
|