Searched defs:work_data (Results 1 – 2 of 2) sorted by relevance
768 auto work_data = allocator.allocate(sizeof(scalar_t)*lwork * batch_size); in apply_cholesky_cusolver_potrf_looped() local1034 auto work_data = allocator.allocate(sizeof(scalar_t) * std::max<int>(1, lwork)); in apply_geqrf() local1111 auto work_data = allocator.allocate(sizeof(scalar_t)*lwork); in apply_ormqr() local1188 auto work_data = allocator.allocate(sizeof(scalar_t)*lwork); in apply_orgqr() local1285 auto work_data = allocator.allocate(sizeof(scalar_t) * lwork); in apply_syevd() local1341 auto work_data = allocator.allocate(sizeof(scalar_t) * lwork); in apply_syevj() local1401 auto work_data = allocator.allocate(sizeof(scalar_t) * lwork); in apply_syevj_batched() local
1040 auto work_data = allocator.allocate(buffer_size); in addmv_out_sparse_csr() local1182 auto work_data = allocator.allocate(buffer_size); in add_out_sparse_csr() local1321 auto work_data = allocator.allocate(buffer_size); in triangular_solve_out_sparse_csr() local