Searched refs:rocm_blas_handle (Results 1 – 1 of 1) sorted by relevance
103 CHECK(wrap::rocblas_create_handle(parent_, &rocm_blas_handle) == in ROCmSolverHandles()106 CHECK(wrap::rocblas_set_stream(parent_, rocm_blas_handle, stream) == in ROCmSolverHandles()112 CHECK(wrap::rocblas_destroy_handle(parent_, rocm_blas_handle) == in ~ROCmSolverHandles()117 rocblas_handle rocm_blas_handle; member156 rocm_blas_handle_ = it->second->rocm_blas_handle; in ROCmSolver()209 rocblas_handle rocm_blas_handle, in TrsmImpl() argument218 TF_RETURN_IF_ROCBLAS_ERROR(solver(gpu_executor, rocm_blas_handle, side, uplo, in TrsmImpl()