Searched defs:unit_diagonal (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | triangular_solve_expander.cc | 367 bool conjugate_a, bool unit_diagonal, in SolveByInvertingDiagonalBlocks()
411 bool conjugate_a, bool unit_diagonal, in SolveDirectly()
468 bool conjugate_a, bool unit_diagonal, int64_t block_size, in BuildTriangularSolve()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | triangular_solve_thunk.cc | 89 se::blas::Side side, se::blas::Diagonal unit_diagonal, in RunTriangulatSolve()
|
| H A D | jitrt_custom_calls.cc | 1422 bool unit_diagonal, TriangularSolveOptions::Transpose transpose_a) const { in operator ()()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | xla_data.proto | 653 bool unit_diagonal = 3; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 4378 bool unit_diagonal, in TriangularSolve()
|