Searched refs:SliceInMinorDims (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 292 auto v = SliceInMinorDims(vs, {0}, {1}); in ComputeWYRepresentation()
293 auto beta = SliceInMinorDims(taus, {0}, {1}); in ComputeWYRepresentation()
354 auto a_block = SliceInMinorDims(a, {i, i}, {m, i + k}); in QRDecomposition()
367 auto a_panel = SliceInMinorDims(a, {i, i + k}, {m, n}); in QRDecomposition()
374 auto q_panel = SliceInMinorDims(q, {0, i}, {m, m}); in QRDecomposition()
385 q = SliceInMinorDims(q, {0, 0}, {m, p}); in QRDecomposition()
386 a = SliceInMinorDims(a, {0, 0}, {p, n}); in QRDecomposition()
|
| D | slicing.h | 31 XlaOp SliceInMinorDims(XlaOp x, absl::Span<const int64> start,
|
| D | matrix.cc | 80 return SliceInMinorDims(result, {std::min<int64>(k, n)}, in GetMatrixDiagonal()
85 return SliceInMinorDims(result, {std::min<int64>(-k, m)}, in GetMatrixDiagonal()
|
| D | svd.cc | 757 Sort({-d, SliceInMinorDims(result.u, {0, 0}, {m, n})}, in SortBySingularValuesAndPostProcessing()
764 Sort({SliceInMinorDims(-d, {0, 0}, {n, n}), result.v}, in SortBySingularValuesAndPostProcessing()
783 SliceInMinorDims(result.u, {0, n}, {m, m})}, in SortBySingularValuesAndPostProcessing()
|
| D | slicing.cc | 21 XlaOp SliceInMinorDims(XlaOp x, absl::Span<const int64> start, in SliceInMinorDims() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 179 auto lhs = SliceInMinorDims(l, {i, 0}, {n, i}); in BuildCholesky()
180 auto rhs = SliceInMinorDims(l, {i, 0}, {i + k, i}); in BuildCholesky()
182 auto before = SliceInMinorDims(a, {i, i}, {n, i + k}); in BuildCholesky()
187 auto x = SliceInMinorDims(a, {i, i}, {i + k, i + k}); in BuildCholesky()
194 auto panel = SliceInMinorDims(a, {i + k, i}, {n, i + k}); in BuildCholesky()
|
| D | triangular_solve_expander.cc | 94 SliceInMinorDims(a, {n - n % block_size, n - n % block_size}, {n, n}); in DiagonalBlocks()
288 MaybeConjugate(Collapse(SliceInMinorDims(inv_diag_blocks, {j, 0, 0}, in SolveWithInvertedDiagonalBlocks()
301 auto b_row = SliceInMinorDims(b, start, end); in SolveWithInvertedDiagonalBlocks()
318 MaybeConjugate(SliceInMinorDims(a, start, end), conjugate_a); in SolveWithInvertedDiagonalBlocks()
|
| /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()
|