Searched refs:unit_diagonal_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | triangular_solve_thunk.cc | 45 unit_diagonal_(options.unit_diagonal() ? se::blas::Diagonal::kUnit in TriangularSolveThunk()
77 << " diagonal=" << se::blas::DiagonalString(unit_diagonal_) in ExecuteOnStream()
100 unit_diagonal_, m_, n_, /*alpha=*/1.0f, in ExecuteOnStream()
110 unit_diagonal_, m_, n_, /*alpha=*/1.0, in ExecuteOnStream()
120 .ThenBlasTrsm(side_, uplo_, transpose_a_, unit_diagonal_, m_, in ExecuteOnStream()
131 .ThenBlasTrsm(side_, uplo_, transpose_a_, unit_diagonal_, m_, in ExecuteOnStream()
|
| D | triangular_solve_thunk.h | 56 const se::blas::Diagonal unit_diagonal_; variable
|