Home
last modified time | relevance | path

Searched refs:new_func_name (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc83 const string& func_name, const string& new_func_name, in FunctionalizeControlFlowForFunction() argument
204 GraphToFunctionDef(*g, new_func_name, &functionalized_fdef)); in FunctionalizeControlFlowForFunction()
205 if (func_name == new_func_name) { in FunctionalizeControlFlowForFunction()
208 fld->ReplaceFunction(new_func_name, functionalized_fdef)); in FunctionalizeControlFlowForFunction()
210 VLOG(2) << "Adding function " << new_func_name; in FunctionalizeControlFlowForFunction()
256 string new_func_name = options.flib_def->UniqueFunctionName( in Run() local
260 func.name(), new_func_name, func.attr(), options.flib_def, flr, in Run()
264 func.set_name(new_func_name); in Run()
Dtf2xla_util.cc198 string new_func_name = in PropagateConstIntoFuncAttr() local
201 GraphToFunctionDef(*func_graph, new_func_name, &replace_fdef)); in PropagateConstIntoFuncAttr()
205 func_attr.set_name(new_func_name); in PropagateConstIntoFuncAttr()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.h92 const NameAttrList& func_name_attrs, const string& new_func_name,
Dextract_outside_compilation_pass.cc1783 string new_func_name = absl::StrCat(canonical_func_name, "_oc"); in ExtractOutsideCompilationForFuncCallNode() local
1788 func, new_func_name, host_func_name, host_compute_core, flr, fld, in ExtractOutsideCompilationForFuncCallNode()
1800 absl::make_unique<NodeDefBuilder>(n->name(), new_func_name, fld); in ExtractOutsideCompilationForFuncCallNode()
2256 const NameAttrList& func_name_attrs, const string& new_func_name, in ExtractOutsideCompilationForFunction() argument
2313 new_func_name); in ExtractOutsideCompilationForFunction()
2422 GraphToFunctionDef(*g, new_func_name, updated_fdef.get())); in ExtractOutsideCompilationForFunction()
2429 if (fld->Find(new_func_name)) { in ExtractOutsideCompilationForFunction()
2430 TF_RETURN_IF_ERROR(fld->ReplaceFunction(new_func_name, *updated_fdef)); in ExtractOutsideCompilationForFunction()
Dencapsulate_subgraphs_pass_test.cc413 const string& new_func_name, const string& oc_cluster, in RecvAtHost() argument
418 new_func_name, "_", oc_cluster); in RecvAtHost()
420 new_func_name, "_", oc_cluster, "_recv"); in RecvAtHost()
433 const string& new_func_name, const string& oc_cluster, in SendFromHost() argument
438 new_func_name, "_", oc_cluster); in SendFromHost()
440 new_func_name, "_", oc_cluster, "_send"); in SendFromHost()
Dextract_outside_compilation_pass_test.cc242 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()
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py587 new_func_name = _get_new_function_name(orig_func_name)
603 output_library_func.signature.name = new_func_name