Searched refs:unit_diagonal (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 352 bool unit_diagonal, int64 block_size, in BuildTriangularSolve() argument
417 if (unit_diagonal) { in BuildTriangularSolve()
463 options.unit_diagonal() ? "unit" : "nonunit"); in ExpandInstruction()
488 transpose_a, conjugate_a, options.unit_diagonal(), in ExpandInstruction()
|
| D | hlo_instructions.cc | 312 options.unit_diagonal() == other_options.unit_diagonal() && in IdenticalSlowPath()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | triangular_solve_thunk.cc | 45 unit_diagonal_(options.unit_diagonal() ? se::blas::Diagonal::kUnit in TriangularSolveThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 747 bool unit_diagonal, in TriangularSolve() argument
750 a.op(), b.op(), left_side, lower, unit_diagonal, in TriangularSolve()
|
| D | local_computation_builder.h | 371 bool lower, bool unit_diagonal, int transpose_a);
|
| D | xla_client.py | 1756 unit_diagonal=False): argument
1764 return self._client.TriangularSolve(a, b, left_side, lower, unit_diagonal,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 815 bool unit_diagonal,
1374 bool unit_diagonal,
|
| D | xla_builder.cc | 3030 bool unit_diagonal, in TriangularSolve() argument
3041 options.set_unit_diagonal(unit_diagonal); in TriangularSolve()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 557 bool unit_diagonal = 3; field
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2523 <b> `TriangularSolve(a, b, left_side, lower, unit_diagonal, transpose_a)` </b>
2538 | `unit_diagonal` | `bool` | if `true`, the diagonal elements of `a` are |
|