Searched refs:loop_cond_func (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1476 NameAttrList* loop_cond_func, FunctionLibraryDefinition* fld, in AddSendLoopPredToLoopCond() argument
1480 const FunctionDef* loop_cond_fdef = fld->Find(loop_cond_func->name()); in AddSendLoopPredToLoopCond()
1483 *loop_cond_fdef, AttrSlice(&loop_cond_func->attr()), fld, &fbody)); in AddSendLoopPredToLoopCond()
1492 loop_cond_func->name(), ": ", in AddSendLoopPredToLoopCond()
1502 loop_cond_func->name()); in AddSendLoopPredToLoopCond()
1528 if (loop_cond_func->name() == cond_xla_func_name) { in AddSendLoopPredToLoopCond()
1530 GraphToFunctionDef(*g, loop_cond_func->name(), &replace_fdef)); in AddSendLoopPredToLoopCond()
1532 fld->ReplaceFunction(loop_cond_func->name(), replace_fdef)); in AddSendLoopPredToLoopCond()
1538 absl::StrCat(loop_cond_func->name(), "_send_pred_added_")); in AddSendLoopPredToLoopCond()
1541 loop_cond_func->set_name(new_name); in AddSendLoopPredToLoopCond()
[all …]
|