Home
last modified time | relevance | path

Searched refs:SolveForSizeOneOrTwoKernel (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dtridiagonal_solve_op_gpu.cu.cc43 __global__ void SolveForSizeOneOrTwoKernel(const int m, in SolveForSizeOneOrTwoKernel() function
233 /*work_element_count=*/1, device, &SolveForSizeOneOrTwoKernel<Scalar>, in SolveForSizeOneOrTwo()
236 TF_CHECK_OK(GpuLaunchKernel(SolveForSizeOneOrTwoKernel<Scalar>, in SolveForSizeOneOrTwo()