Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dtridiagonal_solve_op_gpu.cu.cc247 class TridiagonalSolveOpGpu : public OpKernel { class
249 explicit TridiagonalSolveOpGpu(OpKernelConstruction* context) in TridiagonalSolveOpGpu() function in tensorflow::TridiagonalSolveOpGpu