Searched refs:TridiagonalSolveOpGpu (Results 1 – 1 of 1) sorted by relevance
246 class TridiagonalSolveOpGpu : public OpKernel { class248 explicit TridiagonalSolveOpGpu(OpKernelConstruction* context) in TridiagonalSolveOpGpu() function in tensorflow::TridiagonalSolveOpGpu280 TF_DISALLOW_COPY_AND_ASSIGN(TridiagonalSolveOpGpu);363 REGISTER_LINALG_OP_GPU("TridiagonalSolve", (TridiagonalSolveOpGpu<float>),365 REGISTER_LINALG_OP_GPU("TridiagonalSolve", (TridiagonalSolveOpGpu<double>),367 REGISTER_LINALG_OP_GPU("TridiagonalSolve", (TridiagonalSolveOpGpu<complex64>),369 REGISTER_LINALG_OP_GPU("TridiagonalSolve", (TridiagonalSolveOpGpu<complex128>),