Searched refs:UpdateSliceInMinorDims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 294 y = UpdateSliceInMinorDims(y, v, {0}); in ComputeWYRepresentation()
298 w = UpdateSliceInMinorDims(w, bv, {0}); in ComputeWYRepresentation()
357 a = UpdateSliceInMinorDims(a, qr_block.r, {i, i}); in QRDecomposition()
371 a = UpdateSliceInMinorDims(a, a_panel, {i, i + k}); in QRDecomposition()
378 q = UpdateSliceInMinorDims(q, q_panel, {0, i}); in QRDecomposition()
|
| D | slicing.h | 36 XlaOp UpdateSliceInMinorDims(XlaOp x, XlaOp update,
|
| D | slicing.cc | 69 XlaOp UpdateSliceInMinorDims(XlaOp x, XlaOp update, in UpdateSliceInMinorDims() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 183 a = UpdateSliceInMinorDims(a, before - delta, {i, i}); in BuildCholesky()
189 l = UpdateSliceInMinorDims(l, factorized, {i, i}); in BuildCholesky()
201 l = UpdateSliceInMinorDims(l, update, {i + k, i}); in BuildCholesky()
|