Searched defs:main_diagonal (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 54 XlaOp main_diagonal, in CheckSystemAndReturnNumEquations()
164 XlaOp upper_diagonal, XlaOp main_diagonal, XlaOp lower_diagonal, in CheckMatMulSystemAndReturnShapeParams()
237 XlaOp main_diagonal, in TridiagonalSolverImpl()
280 auto main_diagonal = values[1]; in TridiagonalSolverImpl() local
372 XlaOp main_diagonal, XlaOp upper_diagonal, in TridiagonalSolver()
406 auto main_diagonal = in TridiagonalSolver() local
443 StatusOr<XlaOp> TridiagonalMatMul(XlaOp upper_diagonal, XlaOp main_diagonal, in TridiagonalMatMul()
|
| D | tridiagonal_test.cc | 43 Array3D<float> main_diagonal{{{21, 22, 23}}}; in XLA_TEST_P() local
84 Array<float> main_diagonal = Array<float>({5, 3, 7}, 1); in XLA_TEST_P() local
118 Array3D<float> main_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tridiagonal_ops.cc | 57 auto main_diagonal = ctx->Input(1); in Compile() local
|