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 | svd.cc | 490 d = DynamicUpdateSliceInMinorDims(d, slice_p_new, {p, zero}); in OneSidedJacobiUpdate()
491 d = DynamicUpdateSliceInMinorDims(d, slice_q_new, {q, zero}); in OneSidedJacobiUpdate()
500 d = DynamicUpdateSliceInMinorDims(d, slice_p_new, {zero, p}); in OneSidedJacobiUpdate()
501 d = DynamicUpdateSliceInMinorDims(d, slice_q_new, {zero, q}); in OneSidedJacobiUpdate()
503 d = DynamicUpdateSliceInMinorDims(d, pq_zeros, {p, q}); in OneSidedJacobiUpdate()
504 d = DynamicUpdateSliceInMinorDims(d, pq_zeros, {q, p}); in OneSidedJacobiUpdate()
529 u = DynamicUpdateSliceInMinorDims(u, slice_p_new, {zero, p}); in OneSidedJacobiUpdate()
530 u = DynamicUpdateSliceInMinorDims(u, slice_q_new, {zero, q}); in OneSidedJacobiUpdate()
555 v = DynamicUpdateSliceInMinorDims(v, slice_p_new, {zero, p}); in OneSidedJacobiUpdate()
556 v = DynamicUpdateSliceInMinorDims(v, slice_q_new, {zero, q}); in OneSidedJacobiUpdate()
|
| D | slicing.h | 50 XlaOp DynamicUpdateSliceInMinorDims(XlaOp x, XlaOp update,
|
| D | tridiagonal.cc | 123 return DynamicUpdateSliceInMinorDims( in UpdateEq()
128 return DynamicUpdateSliceInMinorDims(updated, update, /*starts=*/{i}); in UpdateEq()
|
| D | slicing_test.cc | 97 DynamicUpdateSliceInMinorDims(a, b, {x, y}); in XLA_TEST_F()
|
| D | slicing.cc | 149 XlaOp DynamicUpdateSliceInMinorDims(XlaOp x, XlaOp update, in DynamicUpdateSliceInMinorDims() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 315 t = DynamicUpdateSliceInMinorDims(t, z, {j}); in CompactWYRepresentation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 346 static Value DynamicUpdateSliceInMinorDims(Location loc, Value v, Value update, in DynamicUpdateSliceInMinorDims() function
367 return DynamicUpdateSliceInMinorDims(loc, v, update, dus_starts, builder); in UpdateSliceInMinorDims()
6052 w = DynamicUpdateSliceInMinorDims(loc, w, z, {j}, builder); in ComputeWYRepresentation()
|