Searched refs:a_update (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 398 auto a_update = in BuildQrDecomposition() local
401 a_update = BatchDot(yt, a_update, precision); in BuildQrDecomposition()
402 a_panel = a_panel + a_update; in BuildQrDecomposition()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5650 auto a_update = BatchDot(op.getLoc(), w, true, a_panel, false, in matchAndRewrite() local
5652 a_update = BatchDot(op.getLoc(), y, false, a_update, false, in matchAndRewrite()
5654 a_panel = rewriter.create<AddOp>(op.getLoc(), a_panel, a_update); in matchAndRewrite()
|