Searched refs:DynamicUpdateSliceInMinorDims (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | self_adjoint_eig.cc | 136 DynamicUpdateSliceInMinorDims(jacobi_update.w, slice_p_new, {p, zero}); in Update()
138 DynamicUpdateSliceInMinorDims(jacobi_update.w, slice_q_new, {q, zero}); in Update()
147 DynamicUpdateSliceInMinorDims(jacobi_update.w, slice_p_new, {zero, p}); in Update()
149 DynamicUpdateSliceInMinorDims(jacobi_update.w, slice_q_new, {zero, q}); in Update()
158 DynamicUpdateSliceInMinorDims(jacobi_update.w, pq_zeros, {p, q}); in Update()
160 DynamicUpdateSliceInMinorDims(jacobi_update.w, pq_zeros, {q, p}); in Update()
188 DynamicUpdateSliceInMinorDims(jacobi_update.v, slice_p_new, {zero, p}); in Update()
190 DynamicUpdateSliceInMinorDims(jacobi_update.v, slice_q_new, {zero, q}); in Update()
|
| D | qr.cc | 210 a = DynamicUpdateSliceInMinorDims(a, new_x, {j}); in QRBlock()
213 vs = DynamicUpdateSliceInMinorDims( in QRBlock()
216 taus = DynamicUpdateSliceInMinorDims( in QRBlock()
282 w = DynamicUpdateSliceInMinorDims(w, z, {j}); in ComputeWYRepresentation()
283 y = DynamicUpdateSliceInMinorDims(y, v, {j}); in ComputeWYRepresentation()
|
| D | svd.cc | 499 d = DynamicUpdateSliceInMinorDims(d, slice_p_new, {p, zero}); in OneSidedJacobiUpdate()
500 d = DynamicUpdateSliceInMinorDims(d, slice_q_new, {q, zero}); in OneSidedJacobiUpdate()
509 d = DynamicUpdateSliceInMinorDims(d, slice_p_new, {zero, p}); in OneSidedJacobiUpdate()
510 d = DynamicUpdateSliceInMinorDims(d, slice_q_new, {zero, q}); in OneSidedJacobiUpdate()
512 d = DynamicUpdateSliceInMinorDims(d, pq_zeros, {p, q}); in OneSidedJacobiUpdate()
513 d = DynamicUpdateSliceInMinorDims(d, pq_zeros, {q, p}); in OneSidedJacobiUpdate()
538 u = DynamicUpdateSliceInMinorDims(u, slice_p_new, {zero, p}); in OneSidedJacobiUpdate()
539 u = DynamicUpdateSliceInMinorDims(u, slice_q_new, {zero, q}); in OneSidedJacobiUpdate()
564 v = DynamicUpdateSliceInMinorDims(v, slice_p_new, {zero, p}); in OneSidedJacobiUpdate()
565 v = DynamicUpdateSliceInMinorDims(v, slice_q_new, {zero, q}); in OneSidedJacobiUpdate()
|
| D | slicing.h | 43 XlaOp DynamicUpdateSliceInMinorDims(XlaOp x, XlaOp update,
|
| D | slicing_test.cc | 95 DynamicUpdateSliceInMinorDims(a, b, {x, y}); in XLA_TEST_F()
|
| D | slicing.cc | 128 XlaOp DynamicUpdateSliceInMinorDims(XlaOp x, XlaOp update, in DynamicUpdateSliceInMinorDims() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 122 DynamicUpdateSliceInMinorDims(body_l, (a_ip1i - dot_ip1) / l_ii, {i}); in CholeskyUnblocked()
125 body_l = DynamicUpdateSliceInMinorDims(body_l, l_ii, {i, i}); in CholeskyUnblocked()
|
| D | triangular_solve_expander.cc | 343 x = DynamicUpdateSliceInMinorDims(x, x_update, /*starts=*/update_starts); in SolveWithInvertedDiagonalBlocks()
|