Searched refs:CreateWhile32 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 645 static void CreateWhile32(Location loc, int num_iterations, in CreateWhile32() function
5120 CreateWhile32(op.getLoc(), first_dim_size, swap_body_fn, {swaps, indices}, in matchAndRewrite()
5969 CreateWhile32(loc, std::min(m, n), qr_body_fn, {a, *vs, *taus}, in QRBlock()
6074 CreateWhile32(loc, n - 1, body_fn, {w, vs, taus}, &while_output, rewriter); in ComputeWYRepresentation()
|