Searched refs:WhileLoopHelper (Results 1 – 5 of 5) 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 | 326 WhileLoopHelper(while_cond_fn, while_body_fn, values, in HouseHolderBidiagonalization()
680 WhileLoopHelper(while_cond_fn_innermost, while_body_fn_innermost, in WhileLoopFn()
705 WhileLoopHelper(while_cond_fn_inner, while_body_fn_inner, values_inner, in WhileLoopFn()
720 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in WhileLoopFn()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 521 xla::WhileLoopHelper(WhileCondFn(num_boxes, output_size), in Compile()
|