Searched refs:swap_body_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 169 auto swap_body_fn = [&](xla::XlaOp i, in Compile() local
197 xla::ForEachIndex(n, xla::S32, swap_body_fn, {swaps, indices}, in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5078 auto swap_body_fn = [&](Location loc, Value i, ArrayRef<Value> old_values, in matchAndRewrite() local
5120 CreateWhile32(op.getLoc(), first_dim_size, swap_body_fn, {swaps, indices}, in matchAndRewrite()
|