Searched refs:BuildWhile (Results 1 – 2 of 2) sorted by relevance
241 xla::XlaOp BuildWhile(XlaOpKernelContext* ctx, in BuildWhile() function574 BuildWhile(ctx, wrapped_cond, wrapped_body, init, body.input_mapping, in Compile()
147 def BuildWhile(): function159 BuildWhile()