Searched refs:cublas_solver (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 170 std::unique_ptr<CudaSolver> cublas_solver(new CudaSolver(context)); in ComputeMatrix() local
174 cublas_solver->allocate_scoped_tensor( in ComputeMatrix()
176 TransposeWithGeam(context, cublas_solver, rhs.data(), in ComputeMatrix()
180 TransposeWithGeam(context, cublas_solver, temp.flat<Scalar>().data(), in ComputeMatrix()
187 const std::unique_ptr<CudaSolver>& cublas_solver, in TransposeWithGeam() argument
192 cublas_solver->Geam(CUBLAS_OP_T, CUBLAS_OP_N, src_rows, in TransposeWithGeam()
348 std::unique_ptr<CudaSolver> cublas_solver(new CudaSolver(context)); in TransposeLhsForGtsvBatched() local
349 OP_REQUIRES_OK(context, cublas_solver->allocate_scoped_tensor( in TransposeLhsForGtsvBatched()
|