Searched refs:WhileLoopHelper (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | loops.cc | 25 StatusOr<std::vector<XlaOp>> WhileLoopHelper( in WhileLoopHelper() function
117 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in ForEachIndex()
|
| D | loops.h | 49 StatusOr<std::vector<XlaOp>> WhileLoopHelper(
|
| D | self_adjoint_eig.cc | 297 WhileLoopHelper(while_cond_fn_innermost, while_body_fn_innermost, in WhileLoopFn()
320 WhileLoopHelper(while_cond_fn_inner, while_body_fn_inner, values_inner, in WhileLoopFn()
334 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in WhileLoopFn()
|
| D | svd.cc | 318 WhileLoopHelper(while_cond_fn, while_body_fn, values, in HouseHolderBidiagonalization()
670 WhileLoopHelper(while_cond_fn_innermost, while_body_fn_innermost, in WhileLoopFn()
695 WhileLoopHelper(while_cond_fn_inner, while_body_fn_inner, values_inner, in WhileLoopFn()
710 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in WhileLoopFn()
|
| D | math.cc | 772 TF_ASSIGN_OR_RETURN(vals, WhileLoopHelper(cond, body, vals, "igamma", &b)); in IgammaSeries()
905 TF_ASSIGN_OR_RETURN(vals, WhileLoopHelper(cond, body, vals, "igammac", &b)); in IgammacContinuedFraction()
1724 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in LentzThompsonBarnettAlgorithm()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 527 xla::WhileLoopHelper(WhileCondFn(num_boxes, output_size), in Compile()
|