Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtridiagonal_ops.cc27 class TridiagonalSolveOp : public XlaOpKernel { class
29 explicit TridiagonalSolveOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TridiagonalSolveOp() function in tensorflow::__anon6f2d975c0111::TridiagonalSolveOp
/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_solve_op.cc38 class TridiagonalSolveOp : public LinearAlgebraOp<Scalar> { class
44 explicit TridiagonalSolveOp(OpKernelConstruction* context) : Base(context) { in TridiagonalSolveOp() function in tensorflow::TridiagonalSolveOp