Searched refs:a_panel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 397 auto a_panel = SliceInMinorDims(a, {i, i + k}, {m, n}); in BuildQrDecomposition() local
399 BatchDot(MaybeConjugate(y, true), /*transpose_x=*/true, a_panel, in BuildQrDecomposition()
402 a_panel = a_panel + a_update; in BuildQrDecomposition()
403 a = UpdateSliceInMinorDims(a, a_panel, {i, i + k}); in BuildQrDecomposition()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5648 Value a_panel = in matchAndRewrite() local
5650 auto a_update = BatchDot(op.getLoc(), w, true, a_panel, false, in matchAndRewrite()
5654 a_panel = rewriter.create<AddOp>(op.getLoc(), a_panel, a_update); in matchAndRewrite()
5655 a = UpdateSliceInMinorDims(op.getLoc(), a, a_panel, {i, i + k}, in matchAndRewrite()
|