| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 89 const string& new_func_name, bool function_modified) { in UpdateFunctionMap() argument
93 function_modified ? absl::make_optional(new_func_name) : absl::nullopt; in UpdateFunctionMap()
132 const string& func_name, const string& new_func_name,
172 string new_func_name; in FunctionalizeControlFlowForNodeAssociatedFunctions() local
176 new_func_name = GetMappedFunctionName(func_iter); in FunctionalizeControlFlowForNodeAssociatedFunctions()
178 graph, n, fld, associated_function, new_func_name)); in FunctionalizeControlFlowForNodeAssociatedFunctions()
184 new_func_name = in FunctionalizeControlFlowForNodeAssociatedFunctions()
188 func_name, new_func_name, associated_function.attrs(), fld, flr, in FunctionalizeControlFlowForNodeAssociatedFunctions()
190 UpdateFunctionMap(func_map, canonicalized_name, new_func_name, in FunctionalizeControlFlowForNodeAssociatedFunctions()
195 new_func_name, associated_function, graph, fld)); in FunctionalizeControlFlowForNodeAssociatedFunctions()
[all …]
|
| D | tf2xla_util.cc | 198 string new_func_name = in PropagateConstIntoFuncAttr() local
201 GraphToFunctionDef(*func_graph, new_func_name, &replace_fdef)); in PropagateConstIntoFuncAttr()
206 func_attr.set_name(new_func_name); in PropagateConstIntoFuncAttr()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.h | 92 const NameAttrList& func_name_attrs, const string& new_func_name,
|
| D | extract_outside_compilation_pass.cc | 1834 string new_func_name = absl::StrCat(canonical_func_name, "_oc"); in ExtractOutsideCompilationForFuncCallNode() local
1839 func, new_func_name, host_func_name, host_compute_core, flr, fld, in ExtractOutsideCompilationForFuncCallNode()
1851 absl::make_unique<NodeDefBuilder>(n->name(), new_func_name, fld); in ExtractOutsideCompilationForFuncCallNode()
2309 const NameAttrList& func_name_attrs, const string& new_func_name, in ExtractOutsideCompilationForFunction() argument
2366 new_func_name); in ExtractOutsideCompilationForFunction()
2475 GraphToFunctionDef(*g, new_func_name, updated_fdef.get())); in ExtractOutsideCompilationForFunction()
2483 if (fld->Find(new_func_name)) { in ExtractOutsideCompilationForFunction()
2484 TF_RETURN_IF_ERROR(fld->ReplaceFunction(new_func_name, *updated_fdef)); in ExtractOutsideCompilationForFunction()
|
| D | encapsulate_subgraphs_pass_test.cc | 400 const string& new_func_name, const string& oc_cluster, in RecvAtHost() argument
405 new_func_name, "_", oc_cluster); in RecvAtHost()
407 new_func_name, "_", oc_cluster, "_recv"); in RecvAtHost()
420 const string& new_func_name, const string& oc_cluster, in SendFromHost() argument
425 new_func_name, "_", oc_cluster); in SendFromHost()
427 new_func_name, "_", oc_cluster, "_send"); in SendFromHost()
|
| 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()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1883 absl::optional<string> new_func_name, bool* rewritten) { in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef() argument
1892 if (new_func_name) { in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()
1893 rewritten_fdef.mutable_signature()->set_name(*new_func_name); in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()
2319 string new_func_name = in LiftOutsideCompilationOnlyArgsFromCallNode() local
2321 rewritten_fdef.mutable_signature()->set_name(new_func_name); in LiftOutsideCompilationOnlyArgsFromCallNode()
2331 node_def.set_op(new_func_name); in LiftOutsideCompilationOnlyArgsFromCallNode()
2407 string new_func_name = in LiftOutsideCompilationOnlyArgs() local
2410 *function_fbody, flr, fld, lifted_arg_count, new_func_name, in LiftOutsideCompilationOnlyArgs()
2415 f.set_name(new_func_name); in LiftOutsideCompilationOnlyArgs()
2429 string new_func_name = in LiftOutsideCompilationOnlyArgs() local
[all …]
|