Searched refs:qr_body_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 198 auto qr_body_fn = [&](XlaOp j, absl::Span<const XlaOp> values, in QrBlock() local
269 TF_ASSIGN_OR_RETURN(auto values, ForEachIndex(std::min(m, n), S32, qr_body_fn, in QrBlock()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5830 auto qr_body_fn = [&](Location loc, Value j, ArrayRef<Value> old_values, in QRBlock() local
5969 CreateWhile32(loc, std::min(m, n), qr_body_fn, {a, *vs, *taus}, in QRBlock()
|