| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 194 auto vva = BatchDot(v_broadcast, a, precision); in QRBlock()
195 vva = BatchDot(TransposeInMinorDims(v_broadcast), vva, precision); in QRBlock()
274 auto yv = BatchDot(TransposeInMinorDims(y), v, precision); in ComputeWYRepresentation()
276 auto wyv = BatchDot(w, yv, precision); in ComputeWYRepresentation()
368 auto a_update = BatchDot(TransposeInMinorDims(w), a_panel, precision); in QRDecomposition()
369 a_update = BatchDot(y, a_update, precision); in QRDecomposition()
375 auto q_update = BatchDot(q_panel, w, precision); in QRDecomposition()
376 q_update = BatchDot(q_update, TransposeInMinorDims(y), precision); in QRDecomposition()
|
| D | qr_test.cc | 57 xla::BatchDot(result.q, result.r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
87 xla::BatchDot(result.q, result.r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
|
| D | svd.cc | 167 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()
294 BatchDot(BatchDot(u, house_col.v, precision), in HouseHolderBidiagonalization()
303 BatchDot(BatchDot(v, TransposeInMinorDims(house_row.v), precision), in HouseHolderBidiagonalization()
337 BatchDot(BatchDot(values[1], house_col.v, precision), in HouseHolderBidiagonalization()
|
| D | svd_test.cc | 91 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()
|
| 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 | 73 xla::XlaOp BatchDot(
|
| D | matrix_test.cc | 112 BatchDot(l_index, TransposeInMinorDims(row)); in XLA_TEST_F()
|
| D | matrix.cc | 255 XlaOp BatchDot(XlaOp x, XlaOp y, PrecisionConfig::Precision precision) { in BatchDot() function
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cholesky_test.cc | 174 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()
|
| D | triangular_solve_test.cc | 423 BatchDot( in XLA_TEST_F()
470 BatchDot(a_tri, x); in XLA_TEST_P()
472 BatchDot(x, a_tri); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 102 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()
|
| D | triangular_solve_expander.cc | 321 b_row - BatchDot(MaybeTransposeInMinorDims(a_row, transpose_a), x, in SolveWithInvertedDiagonalBlocks()
325 b_row - BatchDot(x, MaybeTransposeInMinorDims(a_row, transpose_a), in SolveWithInvertedDiagonalBlocks()
335 x_update = BatchDot(MaybeTransposeInMinorDims(inv_block, transpose_a), in SolveWithInvertedDiagonalBlocks()
338 x_update = BatchDot(remainder, in SolveWithInvertedDiagonalBlocks()
|
| D | elemental_ir_emitter_test.cc | 64 XLA_TEST_F(ElementalIrEmitterExecutionTest, BatchDot) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_matmul_op.cc | 34 xla::BatchDot(MaybeTransposeInMinorDims( in Compile()
|
| /external/tensorflow/tensorflow/contrib/resampler/ |
| D | BUILD | 108 # TODO(b/74459949) Support BatchDot in CPU backend.
|