| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 166 auto panel = SliceInMinorDims(a, {i, i}, {n, i + k}); in BuildCholesky()
171 auto lhs = SliceInMinorDims(l, {i, 0}, {n, i}); in BuildCholesky()
172 auto rhs = SliceInMinorDims(l, {i, 0}, {i + k, i}); in BuildCholesky()
179 auto x = SliceInMinorDims(panel, {0, 0}, {k, k}); in BuildCholesky()
204 factorized, SliceInMinorDims(panel, {k, 0}, {n - i, k}), in BuildCholesky()
|
| D | triangular_solve_expander.cc | 97 SliceInMinorDims(a, {n - n % block_size, n - n % block_size}, {n, n}); in DiagonalBlocks()
175 MaybeConjugate(Collapse(SliceInMinorDims(inv_diag_blocks, {j, 0, 0}, in SolveWithInvertedDiagonalBlocks()
188 auto b_row = SliceInMinorDims(b, start, end); in SolveWithInvertedDiagonalBlocks()
210 MaybeConjugate(SliceInMinorDims(a, start, end), conjugate_a); in SolveWithInvertedDiagonalBlocks()
439 auto b_row = SliceInMinorDims(b, b_row_start, b_row_end); in SolveDirectly()
447 auto a_chunk = SliceInMinorDims(a, a_start, a_end); in SolveDirectly()
451 SliceInMinorDims(b, {which ? 0 : (j + 1), 0}, {which ? j : m, n}); in SolveDirectly()
457 SliceInMinorDims(b, {0, which ? 0 : (j + 1)}, {m, which ? j : n}); in SolveDirectly()
462 auto a_diag = SliceInMinorDims(a, {j, j}, {j + 1, j + 1}); in SolveDirectly()
|
| D | qr_expander.cc | 379 auto a_block = SliceInMinorDims(a, {i, i}, {m, i + k}); in BuildQrDecomposition()
397 auto a_panel = SliceInMinorDims(a, {i, i + k}, {m, n}); in BuildQrDecomposition()
406 auto q_panel = SliceInMinorDims(q, {0, i}, {m, m}); in BuildQrDecomposition()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 85 q = SliceInMinorDims(q, {0, 0}, {m, p}); in QRDecomposition()
86 r = SliceInMinorDims(r, {0, 0}, {p, n}); in QRDecomposition()
|
| D | slicing.h | 38 XlaOp SliceInMinorDims(XlaOp x, absl::Span<const int64> start,
|
| D | svd_test.cc | 86 u = SliceInMinorDims(u, {0, 0}, {m, n}); in ComputeMatmulUDVT()
88 v = SliceInMinorDims(v, {0, 0}, {n, m}); in ComputeMatmulUDVT()
|
| D | tridiagonal_test.cc | 77 return SliceInMinorDims(operand, /*start=*/{i}, /*end=*/{i + 1}); in XLA_TEST_P()
|
| D | svd.cc | 571 auto w_sliced = SliceInMinorDims(w, {0, 0}, {n, n}); in ComputeToleranceComparison()
747 Sort({d, SliceInMinorDims(result.u, {0, 0}, {m, n})}, in SortBySingularValuesAndPostProcessing()
753 Sort({SliceInMinorDims(d, {0, 0}, {n, n}), result.v}, in SortBySingularValuesAndPostProcessing()
770 SliceInMinorDims(result.u, {0, n}, {m, m})}, in SortBySingularValuesAndPostProcessing()
|
| D | slicing.cc | 41 XlaOp SliceInMinorDims(XlaOp x, absl::Span<const int64> start, in SliceInMinorDims() function
|
| D | matrix.cc | 104 return SliceInMinorDims(result, {0}, in GetMatrixDiagonal()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_svd_op.cc | 75 result.u = xla::SliceInMinorDims(result.u, {0, 0}, {m, p}); in Compile()
78 result.v = xla::SliceInMinorDims(result.v, {0, 0}, {n, p}); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 289 static Value SliceInMinorDims(Location loc, Value v, in SliceInMinorDims() function
5633 SliceInMinorDims(op.getLoc(), a, {i, i}, {m, i + k}, &rewriter); in matchAndRewrite()
5649 SliceInMinorDims(op.getLoc(), a, {i, i + k}, {m, n}, &rewriter); in matchAndRewrite()
5660 SliceInMinorDims(op.getLoc(), q, {0, i}, {m, m}, &rewriter); in matchAndRewrite()
5671 q = SliceInMinorDims(op.getLoc(), q, {0, 0}, {m, p}, &rewriter); in matchAndRewrite()
5672 a = SliceInMinorDims(op.getLoc(), a, {0, 0}, {p, n}, &rewriter); in matchAndRewrite()
6063 auto v = SliceInMinorDims(loc, vs, {0}, {1}, rewriter); in ComputeWYRepresentation()
6064 auto beta = SliceInMinorDims(loc, taus, {0}, {1}, rewriter); in ComputeWYRepresentation()
|