| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | triangular_solve_test.cc | 109 TriangularSolve(a, b, in XLA_TEST_F()
124 TriangularSolve(a, b, in XLA_TEST_F()
145 TriangularSolve(a, b, in XLA_TEST_F()
166 TriangularSolve(a, b, in XLA_TEST_F()
187 TriangularSolve(a, b, in XLA_TEST_F()
208 TriangularSolve(a, b, in XLA_TEST_F()
230 TriangularSolve(a, b, in XLA_TEST_F()
253 TriangularSolve(a, b, in XLA_TEST_F()
271 TriangularSolve(a, b, in XLA_TEST_F()
293 TriangularSolve(a, b, in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MatrixTriangularSolve.pbtxt | 4 name: "linalg.TriangularSolve"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_triangular_solve_op.cc | 33 auto result = xla::TriangularSolve( in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 196 TriangularSolve(factorized, panel, in BuildCholesky()
|
| D | hlo.proto | 204 // Options for TriangularSolve
|
| D | hlo_parser_test.cc | 582 "TriangularSolve", in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 745 LocalOp ComputationBuilder::TriangularSolve(const LocalOp& a, const LocalOp& b, in TriangularSolve() function in xla::swig::ComputationBuilder
749 return xla::TriangularSolve( in TriangularSolve()
|
| D | local_computation_builder.h | 370 LocalOp TriangularSolve(const LocalOp& a, const LocalOp& b, bool left_side,
|
| D | local_computation_builder.i | 399 %unignore xla::swig::ComputationBuilder::TriangularSolve;
|
| D | xla_client.py | 1749 def TriangularSolve(self, member in ComputationBuilder
1764 return self._client.TriangularSolve(a, b, left_side, lower, unit_diagonal,
|
| D | xla_client_test.py | 1197 c.TriangularSolve(c.Constant(a_vals), c.Constant(b_vals), left_side=False,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 814 friend XlaOp TriangularSolve(XlaOp a, XlaOp b, bool left_side, bool lower,
1373 XlaOp TriangularSolve(XlaOp a, XlaOp b, bool left_side, bool lower,
|
| D | xla_builder.cc | 3029 XlaOp TriangularSolve(XlaOp a, XlaOp b, bool left_side, bool lower, in TriangularSolve() function
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2512 ## TriangularSolve section in Operation Semantics
2515 [`XlaBuilder::TriangularSolve`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_…
2523 <b> `TriangularSolve(a, b, left_side, lower, unit_diagonal, transpose_a)` </b>
|