Home
last modified time | relevance | path

Searched refs:TransposeInMinorDims (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dcholesky_test.cc174 BatchDot(input, TransposeInMinorDims(input), PrecisionConfig::HIGHEST); in XLA_TEST_P()
181 verification = BatchDot(cholesky, TransposeInMinorDims(cholesky), in XLA_TEST_P()
184 verification = BatchDot(TransposeInMinorDims(cholesky), cholesky, in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcholesky_expander.cc102 auto diag_dot = BatchDot(row, TransposeInMinorDims(row), precision); in CholeskyUnblocked()
117 auto dot = BatchDot(body_l, TransposeInMinorDims(row), precision); in CholeskyUnblocked()
181 auto delta = BatchDot(lhs, TransposeInMinorDims(rhs), precision); in BuildCholesky()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dqr.cc195 vva = BatchDot(TransposeInMinorDims(v_broadcast), vva, precision); in QRBlock()
274 auto yv = BatchDot(TransposeInMinorDims(y), v, precision); in ComputeWYRepresentation()
368 auto a_update = BatchDot(TransposeInMinorDims(w), a_panel, precision); in QRDecomposition()
376 q_update = BatchDot(q_update, TransposeInMinorDims(y), precision); in QRDecomposition()
Dsvd_test.cc91 return BatchDot(Mul(u, d, broadcast_dims), TransposeInMinorDims(v), in ComputeMatmulUDVT()
171 BatchDot(result.u, TransposeInMinorDims(result.u)); in XLA_TEST_F()
183 BatchDot(result.v, TransposeInMinorDims(result.v), PrecisionConfig::HIGHEST); in XLA_TEST_F()
Dself_adjoint_eig_test.cc111 return BatchDot(vw, TransposeInMinorDims(result.v), in ComputeMatmulVWVt()
189 BatchDot(result.v, TransposeInMinorDims(result.v), PrecisionConfig::HIGHEST); in XLA_TEST_F()
233 BatchDot(TransposeInMinorDims(result.v), result.v), in XLA_TEST_F()
Dsvd.cc167 Sub(a, Mul(beta, BatchDot(BatchDot(a, TransposeInMinorDims(v), precision), in HouseRow()
234 a, Mul(beta, BatchDot(v, BatchDot(TransposeInMinorDims(v), a, precision), in HouseCol()
295 TransposeInMinorDims(house_col.v), precision))); in HouseHolderBidiagonalization()
303 BatchDot(BatchDot(v, TransposeInMinorDims(house_row.v), precision), in HouseHolderBidiagonalization()
338 TransposeInMinorDims(house_col.v), precision))); in HouseHolderBidiagonalization()
847 a = TransposeInMinorDims(a); in SVD()
Dmatrix.h112 xla::XlaOp TransposeInMinorDims(xla::XlaOp x);
Dmatrix.cc339 XlaOp TransposeInMinorDims(XlaOp x) { in TransposeInMinorDims() function
353 return transpose ? TransposeInMinorDims(x) : x; in MaybeTransposeInMinorDims()
Dmatrix_test.cc112 BatchDot(l_index, TransposeInMinorDims(row)); in XLA_TEST_F()
Dself_adjoint_eig.cc439 w_init = w_init + TransposeInMinorDims(w_init) - w_init * v_init; in SelfAdjointEig()