Searched defs:pivots_data (Results 1 – 4 of 4) sorted by relevance
116 auto pivots_data = get_pivots ? pivots.data_ptr<int>() : nullptr; in apply_lu_factor_batched_cublas() local136 auto pivots_data = pivots.const_data_ptr<int>(); in apply_lu_solve_batched_cublas() local
1474 const auto pivots_data = get_pivots ? pivots.data_ptr<int>() : nullptr; in lu_factor_looped_cusolver() local1513 auto pivots_data = pivots.data_ptr<int>(); in lu_solve_looped_cusolver() local
127 …const int32_t* const __restrict__ pivots_data = reinterpret_cast<const int32_t*>(pivots_ptr + offs… in unpack_pivots_cuda_kernel() local
1116 const auto pivots_data = reinterpret_cast<const int32_t*>(pivots_ptr); in unpack_pivots_cpu_kernel() local