Searched refs:MaybeConjugate (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_matmul_op.cc | 33 auto result = xla::BatchDot(MaybeConjugate(ctx->Input(0), adj_x_), adj_x_, in Compile()
34 MaybeConjugate(ctx->Input(1), adj_y_), adj_y_); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 222 auto vva = BatchDot(MaybeConjugate(v_broadcast, true), in QrBlock()
225 a = a - Mul(MaybeConjugate(tau, true), vva, in QrBlock()
326 auto vtv = BatchDot(MaybeConjugate(vs, true), /*transpose_x=*/true, vs, in CompactWYRepresentation()
395 auto yt = BatchDot(y, /*transpose_x=*/false, MaybeConjugate(t, true), in BuildQrDecomposition()
399 BatchDot(MaybeConjugate(y, true), /*transpose_x=*/true, a_panel, in BuildQrDecomposition()
409 BatchDot(q_update, /*transpose_x=*/false, MaybeConjugate(yt, true), in BuildQrDecomposition()
|
| D | cholesky_expander.cc | 99 BatchDot(body_l, false, MaybeConjugate(body_l, true), true, precision); in CholeskyUnblocked()
174 BatchDot(lhs, false, MaybeConjugate(rhs, true), true, precision); in BuildCholesky()
|
| D | triangular_solve_expander.cc | 175 MaybeConjugate(Collapse(SliceInMinorDims(inv_diag_blocks, {j, 0, 0}, in SolveWithInvertedDiagonalBlocks()
210 MaybeConjugate(SliceInMinorDims(a, start, end), conjugate_a); in SolveWithInvertedDiagonalBlocks()
427 a = MaybeConjugate(a, conjugate_a); in SolveDirectly()
535 return unit_diagonal ? b : Div(b, MaybeConjugate(a, conjugate_a)); in BuildTriangularSolve()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.h | 118 xla::XlaOp MaybeConjugate(xla::XlaOp x, bool conjugate);
|
| D | math.cc | 1347 XlaOp MaybeConjugate(XlaOp x, bool conjugate) { in MaybeConjugate() function
|