| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | self_adjoint_eig.cc | 86 auto pqs = DynamicSliceInMinorDims(a, {p, q}, {1, 1}); in SymmetricShurDecomposition2x2()
88 auto ps = DynamicSliceInMinorDims(a, {p, p}, {1, 1}); in SymmetricShurDecomposition2x2()
89 auto qs = DynamicSliceInMinorDims(a, {q, q}, {1, 1}); in SymmetricShurDecomposition2x2()
129 auto slice_p = DynamicSliceInMinorDims(jacobi_update.w, {p, zero}, {1, n}); in Update()
130 auto slice_q = DynamicSliceInMinorDims(jacobi_update.w, {q, zero}, {1, n}); in Update()
140 slice_p = DynamicSliceInMinorDims(jacobi_update.w, {zero, p}, {n, 1}); in Update()
141 slice_q = DynamicSliceInMinorDims(jacobi_update.w, {zero, q}, {n, 1}); in Update()
162 slice_p = DynamicSliceInMinorDims(jacobi_update.v, {zero, p}, {n, 1}); in Update()
163 slice_q = DynamicSliceInMinorDims(jacobi_update.v, {zero, q}, {n, 1}); in Update()
|
| D | svd.cc | 116 XlaOp x = DynamicSliceInMinorDims(a, {i, zero}, {1, n}); in HouseRow()
139 auto x_0j = DynamicSliceInMinorDims(x, {zero, j}, {1, 1}); in HouseRow()
181 XlaOp x = DynamicSliceInMinorDims(a, {zero, j}, {m, 1}); in HouseCol()
205 auto x_0i = DynamicSliceInMinorDims(x, {i, zero}, {1, 1}); in HouseCol()
415 XlaOp a_pp = DynamicSliceInMinorDims(a, {p, p}, {1, 1}); in GetOneSidedJacobiRotation()
416 XlaOp a_pq = DynamicSliceInMinorDims(a, {p, q}, {1, 1}); in GetOneSidedJacobiRotation()
417 XlaOp a_qp = DynamicSliceInMinorDims(a, {q, p}, {1, 1}); in GetOneSidedJacobiRotation()
418 XlaOp a_qq = DynamicSliceInMinorDims(a, {q, q}, {1, 1}); in GetOneSidedJacobiRotation()
484 auto slice_p = DynamicSliceInMinorDims(d, {p, zero}, {1, n}); in OneSidedJacobiUpdate()
485 auto slice_q = DynamicSliceInMinorDims(d, {q, zero}, {1, n}); in OneSidedJacobiUpdate()
[all …]
|
| D | matrix_test.cc | 190 auto l_index = DynamicSliceInMinorDims( in XLA_TEST_F()
211 auto l_index = DynamicSliceInMinorDims( in XLA_TEST_F()
|
| D | slicing_test.cc | 66 DynamicSliceInMinorDims(a, {x, y}, {1, 1}); in XLA_TEST_F()
81 DynamicSliceInMinorDims(a, {index, xla::ConstantR0<int32>(&builder, 0)}, in XLA_TEST_F()
|
| D | sorting.cc | 112 DynamicSliceInMinorDims(input, {start}, {per_partition_size}); in TopKWithPartitions()
114 DynamicSliceInMinorDims(iota_s32, {start}, {per_partition_size}); in TopKWithPartitions()
|
| D | slicing.h | 47 XlaOp DynamicSliceInMinorDims(XlaOp x, absl::Span<const XlaOp> starts,
|
| D | tridiagonal.cc | 112 return DynamicSliceInMinorDims(operand, in Coefficient()
118 return DynamicSliceInMinorDims(operand, in Coefficient()
|
| D | slicing.cc | 130 XlaOp DynamicSliceInMinorDims(XlaOp x, absl::Span<const XlaOp> starts, in DynamicSliceInMinorDims() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 94 XlaOp alpha = Reshape(DynamicSliceInMinorDims(x, {k}, {1}), batch_dims); in House()
204 auto x = DynamicSliceInMinorDims(a, {j}, {1}); in QrBlock()
310 auto yv = DynamicSliceInMinorDims(vtv, {j}, {1}); in CompactWYRepresentation()
|
| D | cholesky_expander.cc | 103 auto l_ii = DynamicSliceInMinorDims(l_square, {i, i}, {1, 1}); in CholeskyUnblocked()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 326 static Value DynamicSliceInMinorDims(Location loc, Value v, in DynamicSliceInMinorDims() function
5719 Value alpha = DynamicSliceInMinorDims(loc, x, {k}, {1}, builder); in House()
5838 auto x = DynamicSliceInMinorDims(loc, a, {j}, {1}, builder); in QRBlock()
6013 auto v = DynamicSliceInMinorDims(loc, vs, {j}, {1}, builder); in ComputeWYRepresentation()
6015 auto beta = DynamicSliceInMinorDims(loc, taus, {j}, {1}, builder); in ComputeWYRepresentation()
|