Home
last modified time | relevance | path

Searched refs:DynamicSliceInMinorDims (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dself_adjoint_eig.cc86 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()
Dsvd.cc121 XlaOp x = DynamicSliceInMinorDims(a, {i, zero}, {1, n}); in HouseRow()
144 auto x_0j = DynamicSliceInMinorDims(x, {zero, j}, {1, 1}); in HouseRow()
187 XlaOp x = DynamicSliceInMinorDims(a, {zero, j}, {m, 1}); in HouseCol()
211 auto x_0i = DynamicSliceInMinorDims(x, {i, zero}, {1, 1}); in HouseCol()
424 XlaOp a_pp = DynamicSliceInMinorDims(a, {p, p}, {1, 1}); in GetOneSidedJacobiRotation()
425 XlaOp a_pq = DynamicSliceInMinorDims(a, {p, q}, {1, 1}); in GetOneSidedJacobiRotation()
426 XlaOp a_qp = DynamicSliceInMinorDims(a, {q, p}, {1, 1}); in GetOneSidedJacobiRotation()
427 XlaOp a_qq = DynamicSliceInMinorDims(a, {q, q}, {1, 1}); in GetOneSidedJacobiRotation()
493 auto slice_p = DynamicSliceInMinorDims(d, {p, zero}, {1, n}); in OneSidedJacobiUpdate()
494 auto slice_q = DynamicSliceInMinorDims(d, {q, zero}, {1, n}); in OneSidedJacobiUpdate()
[all …]
Dslicing_test.cc64 DynamicSliceInMinorDims(a, {x, y}, {1, 1}); in XLA_TEST_F()
79 DynamicSliceInMinorDims(a, {index, xla::ConstantR0<int32>(&builder, 0)}, in XLA_TEST_F()
Dqr.cc89 XlaOp alpha = Reshape(DynamicSliceInMinorDims(x, {k}, {1}), batch_dims); in House()
183 auto x = DynamicSliceInMinorDims(a, {j}, {1}); in QRBlock()
269 auto v = DynamicSliceInMinorDims(vs, {j}, {1}); in ComputeWYRepresentation()
271 auto beta = DynamicSliceInMinorDims(taus, {j}, {1}); in ComputeWYRepresentation()
Dmatrix_test.cc110 auto l_index = DynamicSliceInMinorDims( in XLA_TEST_F()
131 auto l_index = DynamicSliceInMinorDims( in XLA_TEST_F()
Dslicing.h40 XlaOp DynamicSliceInMinorDims(XlaOp x, absl::Span<const XlaOp> starts,
Dsvd_test.cc82 u = DynamicSliceInMinorDims(u, {zero, zero}, {m, n}); in ComputeMatmulUDVT()
84 v = DynamicSliceInMinorDims(v, {zero, zero}, {n, m}); in ComputeMatmulUDVT()
Dslicing.cc109 XlaOp DynamicSliceInMinorDims(XlaOp x, absl::Span<const XlaOp> starts, in DynamicSliceInMinorDims() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dcholesky_expander.cc97 auto l_i = DynamicSliceInMinorDims(body_l, {i, zero}, {1, n}); in CholeskyUnblocked()
100 auto a_ii = DynamicSliceInMinorDims(body_a, {i, i}, {1, 1}); in CholeskyUnblocked()
109 auto a_0i = DynamicSliceInMinorDims(body_a, {i}, {1}); in CholeskyUnblocked()