Lines Matching refs:buffer_index
68 for (size_t buffer_index = 0; buffer_index < num_buffers; buffer_index++) { in f32_spmm() local
92 nmap[buffer_index * nmap_elements + i] += 1; in f32_spmm()
109 nmap[buffer_index * nmap_elements + i] += 1; in f32_spmm()
118 a_offsets[buffer_index] = first_j * mc; in f32_spmm()
134 size_t buffer_index = 0; in f32_spmm() local
142 buffer_index = (buffer_index + 1) % num_buffers; in f32_spmm()
146 a.data() + a_offsets[buffer_index], in f32_spmm()
147 w.data() + buffer_index * w_elements, in f32_spmm()
148 dmap.data() + buffer_index * dmap_elements, in f32_spmm()
149 nmap.data() + buffer_index * nmap_elements, in f32_spmm()
150 c.data() + buffer_index * c_elements, mc * sizeof(float), in f32_spmm()