Searched refs:MaybeTransposeInMinorDims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_matmul_op.cc | 34 xla::BatchDot(MaybeTransposeInMinorDims( in Compile()
36 MaybeTransposeInMinorDims( in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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()
339 MaybeTransposeInMinorDims(inv_block, transpose_a), in SolveWithInvertedDiagonalBlocks()
|
| D | cholesky_expander.cc | 238 XlaOp l = BuildCholesky(MaybeTransposeInMinorDims(a, !options.lower()), in ExpandInstruction()
241 MaybeTransposeInMinorDims(l, !options.lower()); in ExpandInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.h | 116 xla::XlaOp MaybeTransposeInMinorDims(xla::XlaOp x, bool transpose);
|
| D | matrix.cc | 352 XlaOp MaybeTransposeInMinorDims(XlaOp x, bool transpose) { in MaybeTransposeInMinorDims() function
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | triangular_solve_test.cc | 467 a_tri = MaybeTransposeInMinorDims( in XLA_TEST_P()
|