Searched refs:UpdateSliceInMinorDims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.h | 43 XlaOp UpdateSliceInMinorDims(XlaOp x, XlaOp update,
|
| D | slicing.cc | 90 XlaOp UpdateSliceInMinorDims(XlaOp x, XlaOp update, in UpdateSliceInMinorDims() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 198 l = UpdateSliceInMinorDims(l, factorized, {i, i}); in BuildCholesky()
209 l = UpdateSliceInMinorDims(l, update, {i + k, i}); in BuildCholesky()
|
| D | qr_expander.cc | 386 a = UpdateSliceInMinorDims(a, qr_block.a, {i, i}); in BuildQrDecomposition()
403 a = UpdateSliceInMinorDims(a, a_panel, {i, i + k}); in BuildQrDecomposition()
412 q = UpdateSliceInMinorDims(q, q_panel, {0, i}); in BuildQrDecomposition()
|
| D | triangular_solve_expander.cc | 466 b = UpdateSliceInMinorDims(b, b_row, b_row_start); in SolveDirectly()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 359 static Value UpdateSliceInMinorDims(Location loc, Value v, Value update, in UpdateSliceInMinorDims() function
5638 a = UpdateSliceInMinorDims(op.getLoc(), a, r_block, {i, i}, &rewriter); in matchAndRewrite()
5655 a = UpdateSliceInMinorDims(op.getLoc(), a, a_panel, {i, i + k}, in matchAndRewrite()
5666 q = UpdateSliceInMinorDims(op.getLoc(), q, q_panel, {i}, &rewriter); in matchAndRewrite()
6071 w = UpdateSliceInMinorDims(loc, w, bv, {0}, rewriter); in ComputeWYRepresentation()
|