Home
last modified time | relevance | path

Searched refs:new_fdef (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc365 FunctionDef new_fdef; in MaybeRewriteWhileNode() local
368 TF_RETURN_IF_ERROR(GraphToFunctionDef(*fbody->graph, new_name, &new_fdef)); in MaybeRewriteWhileNode()
370 fld->AddFunctionDef(new_fdef, fld->GetStackTraces(attr_value.name()))); in MaybeRewriteWhileNode()
456 FunctionDef new_fdef; in MaybeRewriteIfNode() local
459 TF_RETURN_IF_ERROR(GraphToFunctionDef(*fbody->graph, new_name, &new_fdef)); in MaybeRewriteIfNode()
464 TF_RETURN_IF_ERROR(fld->AddFunctionDef(new_fdef, stack_traces)); in MaybeRewriteIfNode()
Dfunctionalize_control_flow.cc123 const FunctionDef* new_fdef = fld->Find(func_name); in AddFunctionDefToGraphLibrary() local
124 DCHECK(new_fdef != nullptr); in AddFunctionDefToGraphLibrary()
126 *(fdef_lib.add_function()) = *new_fdef; in AddFunctionDefToGraphLibrary()
Dtf2xla_util.cc917 FunctionDef new_fdef; in RewriteTensorListWithConstElement() local
921 GraphToFunctionDef(*bwd_fbody->graph, new_name, &new_fdef)); in RewriteTensorListWithConstElement()
922 TF_RETURN_IF_ERROR(fld->AddFunctionDef(new_fdef)); in RewriteTensorListWithConstElement()