Searched refs:while_cond_fn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | loops.cc | 87 auto while_cond_fn = [&](absl::Span<const XlaOp> values, in ForEachIndex() local
117 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in ForEachIndex()
|
| D | svd.cc | 275 auto while_cond_fn = [&](absl::Span<const XlaOp> values, in HouseHolderBidiagonalization() local
326 WhileLoopHelper(while_cond_fn, while_body_fn, values, in HouseHolderBidiagonalization()
604 auto while_cond_fn = [&](absl::Span<const XlaOp> values, in WhileLoopFn() local
720 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in WhileLoopFn()
|
| D | self_adjoint_eig.cc | 225 auto while_cond_fn = [&](absl::Span<const XlaOp> values, in WhileLoopFn() local
334 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in WhileLoopFn()
|