Searched refs:qr_body_fn (Results 1 – 1 of 1) sorted by relevance
176 auto qr_body_fn = [&](XlaOp j, absl::Span<const XlaOp> values, in QRBlock() local226 TF_ASSIGN_OR_RETURN(auto values, ForEachIndex(std::min(m, n), S32, qr_body_fn, in QRBlock()