Home
last modified time | relevance | path

Searched refs:while_body_fn (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dloops.cc92 auto while_body_fn = in ForEachIndex() local
117 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in ForEachIndex()
Dsvd.cc280 auto while_body_fn = in HouseHolderBidiagonalization() local
326 WhileLoopHelper(while_cond_fn, while_body_fn, values, in HouseHolderBidiagonalization()
619 auto while_body_fn = in WhileLoopFn() local
720 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in WhileLoopFn()
Dself_adjoint_eig.cc240 auto while_body_fn = in WhileLoopFn() local
334 TF_ASSIGN_OR_RETURN(values, WhileLoopHelper(while_cond_fn, while_body_fn, in WhileLoopFn()