Searched refs:TransposeInMinorDims (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cholesky_test.cc | 239 BatchDot(input, TransposeInMinorDims(input), PrecisionConfig::HIGHEST); in XLA_TEST_P()
246 verification = BatchDot(cholesky, TransposeInMinorDims(cholesky), in XLA_TEST_P()
249 verification = BatchDot(TransposeInMinorDims(cholesky), cholesky, in XLA_TEST_P()
283 auto matrix = BatchDot(input, TransposeInMinorDims(Conj(input)), in XLA_TEST_P()
291 verification = BatchDot(cholesky, TransposeInMinorDims(Conj(cholesky)), in XLA_TEST_P()
294 verification = BatchDot(TransposeInMinorDims(Conj(cholesky)), cholesky, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 95 return BatchDot(Mul(u, d, broadcast_dims), TransposeInMinorDims(v), in ComputeMatmulUDVT()
156 BatchDot(result.u, TransposeInMinorDims(result.u)); in XLA_TEST_F()
168 BatchDot(result.v, TransposeInMinorDims(result.v), PrecisionConfig::HIGHEST); in XLA_TEST_F()
|
| D | self_adjoint_eig_test.cc | 111 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()
|
| D | matrix.h | 135 xla::XlaOp TransposeInMinorDims(xla::XlaOp x);
|
| D | matrix_test.cc | 192 BatchDot(l_index, TransposeInMinorDims(row)); in XLA_TEST_F()
|
| D | matrix.cc | 697 XlaOp TransposeInMinorDims(XlaOp x) { in TransposeInMinorDims() function
711 return transpose ? TransposeInMinorDims(x) : x; in MaybeTransposeInMinorDims()
|
| D | self_adjoint_eig.cc | 439 w_init = w_init + TransposeInMinorDims(w_init) - w_init * v_init; in SelfAdjointEig()
|
| D | svd.cc | 834 a = TransposeInMinorDims(a); in SVD()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_inverse_op.cc | 49 qr.ValueOrDie().r, xla::TransposeInMinorDims(qr.ValueOrDie().q), in Compile()
|
| D | matrix_solve_op.cc | 53 qr.ValueOrDie().r, xla::TransposeInMinorDims(qr.ValueOrDie().q), in Compile()
|
| D | extract_image_patches_op.cc | 164 conv = xla::TransposeInMinorDims(xla::Reshape(conv, conv_dims)); in Compile()
|