Searched refs:conjugate_a (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.h | 43 bool conjugate_a, bool unit_diagonal,
53 bool transpose_a, bool conjugate_a, bool unit_diagonal,
57 bool transpose_a, bool conjugate_a,
|
| D | triangular_solve_expander.cc | 136 bool transpose_a, bool conjugate_a, in SolveWithInvertedDiagonalBlocks() argument
178 conjugate_a); in SolveWithInvertedDiagonalBlocks()
210 MaybeConjugate(SliceInMinorDims(a, start, end), conjugate_a); in SolveWithInvertedDiagonalBlocks()
374 bool conjugate_a, bool unit_diagonal, in SolveByInvertingDiagonalBlocks() argument
404 lower, transpose_a, conjugate_a, in SolveByInvertingDiagonalBlocks()
418 bool conjugate_a, bool unit_diagonal, in SolveDirectly() argument
427 a = MaybeConjugate(a, conjugate_a); in SolveDirectly()
475 bool conjugate_a, bool unit_diagonal, int64 block_size, in BuildTriangularSolve() argument
535 return unit_diagonal ? b : Div(b, MaybeConjugate(a, conjugate_a)); in BuildTriangularSolve()
541 return SolveDirectly(a, b, left_side, lower, transpose_a, conjugate_a, in BuildTriangularSolve()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 553 bool conjugate_a = this->conjugate_a_; in Compute() local
556 if (conjugate_a) { in Compute()
560 conjugate_a = false; in Compute()
572 conjugate_a); in Compute()
914 CSRSparseMatrixMatVec(bool transpose_a, bool conjugate_a) in CSRSparseMatrixMatVec() argument
915 : transA_(TransposeAndConjugateToGpuSparseOp(transpose_a, conjugate_a, in CSRSparseMatrixMatVec()
|