Home
last modified time | relevance | path

Searched defs:main_diagonal (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtridiagonal.cc52 XlaOp main_diagonal, in CheckSystemAndReturnNumEquations()
150 XlaOp main_diagonal, in TridiagonalSolverImpl()
193 auto main_diagonal = values[1]; in TridiagonalSolverImpl() local
285 XlaOp main_diagonal, XlaOp upper_diagonal, in TridiagonalSolver()
319 auto main_diagonal = in TridiagonalSolver() local
Dtridiagonal_test.cc48 Array3D<float> main_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local