Searched refs:while_cond_fn (Results 1 – 4 of 4) 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 | 269 auto while_cond_fn = [&](absl::Span<const XlaOp> values, in HouseHolderBidiagonalization() local
318 WhileLoopHelper(while_cond_fn, while_body_fn, values, in HouseHolderBidiagonalization()
594 auto while_cond_fn = [&](absl::Span<const XlaOp> values, in WhileLoopFn() local
710 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()
|
| D | math.cc | 1653 auto while_cond_fn = [num_iterations]( in LentzThompsonBarnettAlgorithm() local
1724 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in LentzThompsonBarnettAlgorithm()
|