Searched refs:backprop_execution_pred (Results 1 – 1 of 1) sorted by relevance
98 Output* backprop_execution_pred) { in AddBackPropLoopCounter() argument126 /* create_while_ctx */ false, backprop_execution_pred)); in AddBackPropLoopCounter()138 const Output& backprop_execution_pred, in AddWhileGradientLoop() argument151 CondGraphBuilderFn cond_fn = [backprop_execution_pred]( in AddWhileGradientLoop()155 *output = backprop_execution_pred; in AddWhileGradientLoop()