| /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()
|
| D | triangular_solve_test.cc | 433 BatchDot( in XLA_TEST_F()
487 BatchDot(a_tri, x); in XLA_TEST_P()
489 BatchDot(x, a_tri); in XLA_TEST_P()
|
| D | dot_operation_test.cc | 1304 auto batch_dot = BatchDot(x, y); in XLA_TEST_P()
1326 INSTANTIATE_TEST_SUITE_P(BatchDot, BatchDotTest,
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr_test.cc | 60 xla::BatchDot(result.q, result.r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
89 xla::BatchDot(result.q, result.r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
121 xla::BatchDot(result.q, result.r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
|
| D | matrix.h | 83 xla::XlaOp BatchDot(
86 xla::XlaOp BatchDot(
|
| D | svd.cc | 161 result.a = Sub(a, Mul(beta, BatchDot(BatchDot(a, false, v, true, precision), in HouseRow()
228 a, Mul(beta, BatchDot(v, false, BatchDot(v, true, a, false, precision), in HouseCol()
288 Mul(house_col.beta, BatchDot(BatchDot(u, house_col.v, precision), in HouseHolderBidiagonalization()
295 BatchDot(BatchDot(v, false, house_row.v, true, precision), in HouseHolderBidiagonalization()
328 BatchDot(BatchDot(values[1], house_col.v, precision), in HouseHolderBidiagonalization()
|
| 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.cc | 539 XlaOp BatchDot(XlaOp x, XlaOp y, PrecisionConfig::Precision precision) { in BatchDot() function
540 return BatchDot(x, false, y, false, precision); in BatchDot()
543 XlaOp BatchDot(XlaOp x, bool transpose_x, XlaOp y, bool transpose_y, in BatchDot() function
|
| D | matrix_test.cc | 192 BatchDot(l_index, TransposeInMinorDims(row)); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 222 auto vva = BatchDot(MaybeConjugate(v_broadcast, true), in QrBlock()
224 vva = BatchDot(v_broadcast, true, vva, false, precision); in QrBlock()
313 auto z = BatchDot(t, yv, precision); in CompactWYRepresentation()
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()
401 a_update = BatchDot(yt, a_update, precision); in BuildQrDecomposition()
407 auto q_update = BatchDot(q_panel, y, precision); in BuildQrDecomposition()
409 BatchDot(q_update, /*transpose_x=*/false, MaybeConjugate(yt, true), in BuildQrDecomposition()
|
| D | triangular_solve_expander.cc | 212 remainder = b_row - BatchDot(a_row, transpose_a, x, false, precision); in SolveWithInvertedDiagonalBlocks()
214 remainder = b_row - BatchDot(x, false, a_row, transpose_a, precision); in SolveWithInvertedDiagonalBlocks()
221 BatchDot(inv_block, transpose_a, remainder, false, precision); in SolveWithInvertedDiagonalBlocks()
224 BatchDot(remainder, false, inv_block, transpose_a, precision); in SolveWithInvertedDiagonalBlocks()
452 b_row = b_row - BatchDot(a_chunk, /*transpose_x=*/transpose_a, b_chunk, in SolveDirectly()
458 b_row = b_row - BatchDot(b_chunk, /*transpose_x=*/false, a_chunk, in SolveDirectly()
|
| 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 | elemental_ir_emitter_test.cc | 99 XLA_TEST_F(ElementalIrEmitterExecutionTest, BatchDot) { in XLA_TEST_F() argument
|
| /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()
|
| D | matrix_solve_op.cc | 60 xla::BatchDot(inv, adjoint_, rhs, in Compile()
|
| D | matmul_op.cc | 87 ctx->SetOutput(0, xla::BatchDot(a, transpose_a_, b, transpose_b_)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 539 Value BatchDot(Location loc, Value lhs, bool transpose_lhs, Value rhs, in BatchDot() function
5650 auto a_update = BatchDot(op.getLoc(), w, true, a_panel, false, in matchAndRewrite()
5652 a_update = BatchDot(op.getLoc(), y, false, a_update, false, in matchAndRewrite()
5661 Value q_update = BatchDot(op.getLoc(), q_panel, false, w, false, in matchAndRewrite()
5663 q_update = BatchDot(op.getLoc(), q_update, false, y, true, in matchAndRewrite()
5857 auto vva = BatchDot(loc, v_broadcast, false, a, false, num_batch_dims, in QRBlock()
5859 vva = BatchDot(loc, v_broadcast, true, vva, false, num_batch_dims, in QRBlock()
6037 auto yv = BatchDot(loc, y, true, v, false, batch_dims.size(), precision, in ComputeWYRepresentation()
6040 auto wyv = BatchDot(loc, w, false, yv, false, batch_dims.size(), in ComputeWYRepresentation()
|