Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtridiagonal_ops.cc27 class TridiagonalSolveOp : public XlaOpKernel { class
29 explicit TridiagonalSolveOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TridiagonalSolveOp() function in tensorflow::__anon693dbe8b0111::TridiagonalSolveOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dtridiagonal_solve_op.cc40 class TridiagonalSolveOp : public LinearAlgebraOp<Scalar> { class
46 explicit TridiagonalSolveOp(OpKernelConstruction* context) : Base(context) { in TridiagonalSolveOp() function in tensorflow::TridiagonalSolveOp