Searched refs:replace_fdef (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 197 FunctionDef replace_fdef; in PropagateConstIntoFuncAttr() local
201 GraphToFunctionDef(*func_graph, new_func_name, &replace_fdef)); in PropagateConstIntoFuncAttr()
203 replace_fdef, lookup_fld->GetStackTraces(func_attr.name()))); in PropagateConstIntoFuncAttr()
211 replace_fdef, lookup_fld->GetStackTraces(func_attr.name()))); in PropagateConstIntoFuncAttr()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1383 FunctionDef replace_fdef; in ReplaceKeyPlaceholderWithArgNode() local
1385 *g, func_name, HostGraphControlRetMapping, &replace_fdef)); in ReplaceKeyPlaceholderWithArgNode()
1386 TF_RETURN_IF_ERROR(fld->ReplaceFunction(func_name, replace_fdef)); in ReplaceKeyPlaceholderWithArgNode()
1527 FunctionDef replace_fdef; in AddSendLoopPredToLoopCond() local
1530 GraphToFunctionDef(*g, loop_cond_func->name(), &replace_fdef)); in AddSendLoopPredToLoopCond()
1532 fld->ReplaceFunction(loop_cond_func->name(), replace_fdef)); in AddSendLoopPredToLoopCond()
1539 TF_RETURN_IF_ERROR(GraphToFunctionDef(*g, new_name, &replace_fdef)); in AddSendLoopPredToLoopCond()
1540 TF_RETURN_IF_ERROR(fld->AddFunctionDef(replace_fdef)); in AddSendLoopPredToLoopCond()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 2949 FunctionDef replace_fdef; in ProcessHeadTailOutsideCompilation() local
2951 GraphToFunctionDef(*xla_graph, xla_func_name, &replace_fdef)); in ProcessHeadTailOutsideCompilation()
2952 TF_RETURN_IF_ERROR(fld->ReplaceFunction(xla_func_name, replace_fdef)); in ProcessHeadTailOutsideCompilation()
|