Searched refs:q_panel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 406 auto q_panel = SliceInMinorDims(q, {0, i}, {m, m}); in BuildQrDecomposition() local
407 auto q_update = BatchDot(q_panel, y, precision); in BuildQrDecomposition()
411 q_panel = q_panel + q_update; in BuildQrDecomposition()
412 q = UpdateSliceInMinorDims(q, q_panel, {0, i}); in BuildQrDecomposition()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5659 Value q_panel = in matchAndRewrite() local
5661 Value q_update = BatchDot(op.getLoc(), q_panel, false, w, false, in matchAndRewrite()
5665 q_panel = rewriter.create<AddOp>(op.getLoc(), q_panel, q_update); in matchAndRewrite()
5666 q = UpdateSliceInMinorDims(op.getLoc(), q, q_panel, {i}, &rewriter); in matchAndRewrite()
|