Home
last modified time | relevance | path

Searched refs:unit_diagonal (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc369 bool unit_diagonal, int64 block_size, in BuildTriangularSolve() argument
434 if (unit_diagonal) { in BuildTriangularSolve()
480 options.unit_diagonal() ? "unit" : "nonunit"); in ExpandInstruction()
505 transpose_a, conjugate_a, options.unit_diagonal(), in ExpandInstruction()
Dhlo_instructions.cc313 options.unit_diagonal() == other_options.unit_diagonal() && in IdenticalSlowPath()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtriangular_solve_thunk.cc45 unit_diagonal_(options.unit_diagonal() ? se::blas::Diagonal::kUnit in TriangularSolveThunk()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dops.mlir585 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
593 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
601 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
609 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
617 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
625 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
633 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1589 unit_diagonal=False): argument
1599 return ops.TriangularSolve(a, b, left_side, lower, unit_diagonal, transpose)
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir840 …%arg1) {left_side = true, lower = true, transpose_a = "NO_TRANSPOSE", unit_diagonal = true} : (ten…
847 …(f32[4,4] [[ARG_A]], f32[4,3] [[ARG_B]]), left_side=true, lower=true, unit_diagonal=true, transpos…
Dimport.hlotxt756 // CHECK-SAME: unit_diagonal = true
757 …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/
Dhlo_function_importer.cc463 instruction->triangular_solve_options().unit_diagonal()))); in ImportInstruction()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h854 bool unit_diagonal,
1463 bool unit_diagonal,
Dxla_builder.cc3095 bool unit_diagonal, in TriangularSolve() argument
3106 options.set_unit_diagonal(unit_diagonal); in TriangularSolve()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dhlo_ops.td1150 BoolAttr:$unit_diagonal,
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto570 bool unit_diagonal = 3; field
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2719 <b> `TriangularSolve(a, b, left_side, lower, unit_diagonal, transpose_a)` </b>
2734 | `unit_diagonal` | `bool` | if `true`, the diagonal elements of `a` are |