Searched refs:TransposeWithGeam (Results 1 – 1 of 1) sorted by relevance
176 TransposeWithGeam(context, cublas_solver, rhs.data(), in ComputeMatrix()180 TransposeWithGeam(context, cublas_solver, temp.flat<Scalar>().data(), in ComputeMatrix()186 void TransposeWithGeam(OpKernelContext* context, in TransposeWithGeam() function in tensorflow::TridiagonalSolveOpGpuLinalg