| /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,
58 bool unit_diagonal, int64 block_size,
|
| D | triangular_solve_expander.cc | 374 bool conjugate_a, bool unit_diagonal, in SolveByInvertingDiagonalBlocks() argument
384 if (unit_diagonal) { in SolveByInvertingDiagonalBlocks()
418 bool conjugate_a, bool unit_diagonal, in SolveDirectly() argument
461 if (!unit_diagonal) { 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()
542 unit_diagonal, precision); in BuildTriangularSolve()
545 conjugate_a, unit_diagonal, in BuildTriangularSolve()
572 options.unit_diagonal() ? "unit" : "nonunit"); in ExpandInstruction()
597 transpose_a, conjugate_a, options.unit_diagonal(), in ExpandInstruction()
|
| D | hlo_instructions.cc | 368 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()
|
| D | ir_emitter_unnested.cc | 1649 options.set_unit_diagonal(triangular_solve_op.unit_diagonal()); in EmitTriangularSolveFromMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | ops.mlir | 884 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
892 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
900 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
908 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
916 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
924 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
932 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
|
| D | lhlo_ops.mlir | 873 unit_diagonal = true}
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 944 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
951 …(f32[4,4] [[ARG_A]], f32[4,3] [[ARG_B]]), left_side=true, lower=true, unit_diagonal=true, transpos…
1194 …arg1) {left_side = false, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = false} : (te…
|
| D | import.hlotxt | 906 // CHECK-SAME: unit_diagonal = true
907 …g_0.1, f32[4,3] %Arg_1.2), left_side=true, lower=true, transpose_a=NO_TRANSPOSE, unit_diagonal=true
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 451 builder_.getBoolAttr(options.unit_diagonal()), in TriangularSolveInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 637 BoolAttr:$unit_diagonal,
|
| D | hlo_ops.td | 1233 BoolAttr:$unit_diagonal,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 649 instruction->triangular_solve_options().unit_diagonal()))); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 679 // CHECK-SAME: unit_diagonal = false
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 1206 bool unit_diagonal,
1968 bool unit_diagonal,
|
| D | xla_builder.cc | 4422 bool unit_diagonal, in TriangularSolve() argument
4431 options.set_unit_diagonal(unit_diagonal); in TriangularSolve()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 605 bool unit_diagonal = 3; field
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1308 builder_.getBoolAttr(options.unit_diagonal())); in EmitTriangularSolveOp()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1350 unit_diagonal=False)
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2829 <b> `TriangularSolve(a, b, left_side, lower, unit_diagonal, transpose_a)` </b>
2844 | `unit_diagonal` | `bool` | if `true`, the diagonal elements of `a` are |
|