Home
last modified time | relevance | path

Searched defs:lower_diagonal (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dtridiagonal.cc53 StatusOr<int64_t> CheckSystemAndReturnNumEquations(XlaOp lower_diagonal, in CheckSystemAndReturnNumEquations()
164 XlaOp upper_diagonal, XlaOp main_diagonal, XlaOp lower_diagonal, in CheckMatMulSystemAndReturnShapeParams()
236 StatusOr<XlaOp> TridiagonalSolverImpl<kThomas>(XlaOp lower_diagonal, in TridiagonalSolverImpl()
279 auto lower_diagonal = values[0]; in TridiagonalSolverImpl() local
371 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp lower_diagonal, in TridiagonalSolver()
409 auto lower_diagonal = in TridiagonalSolver() local
444 XlaOp lower_diagonal, XlaOp rhs) { in TridiagonalMatMul()
H A Dtridiagonal_test.cc45 Array3D<float> lower_diagonal{{{999, 10, 100}}}; in XLA_TEST_P() local
85 Array<float> lower_diagonal = Array<float>({5, 3, 7}, 1); in XLA_TEST_P() local
117 Array3D<float> lower_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtridiagonal_ops.cc58 auto lower_diagonal = ctx->Input(2); in Compile() local