Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtridiagonal_ops.cc52 class TridiagonalMatMulOp : public XlaOpKernel { class
54 explicit TridiagonalMatMulOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TridiagonalMatMulOp() function in tensorflow::__anon5542a90b0111::TridiagonalMatMulOp
/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_matmul_op.cc30 class TridiagonalMatMulOp : public LinearAlgebraOp<Scalar> { class
34 explicit TridiagonalMatMulOp(OpKernelConstruction* context) : Base(context) {} in TridiagonalMatMulOp() function in tensorflow::TridiagonalMatMulOp