Lines Matching refs:buffer_index
71 for (size_t buffer_index = 0; buffer_index < num_buffers; buffer_index++) { in f16_spmm() local
95 nmap[buffer_index * nmap_elements + i] += 1; in f16_spmm()
112 nmap[buffer_index * nmap_elements + i] += 1; in f16_spmm()
121 a_offsets[buffer_index] = first_j * mc; in f16_spmm()
137 size_t buffer_index = 0; in f16_spmm() local
145 buffer_index = (buffer_index + 1) % num_buffers; in f16_spmm()
149 a.data() + a_offsets[buffer_index], in f16_spmm()
150 w.data() + buffer_index * w_elements, in f16_spmm()
151 dmap.data() + buffer_index * dmap_elements, in f16_spmm()
152 nmap.data() + buffer_index * nmap_elements, in f16_spmm()
153 c.data() + buffer_index * c_elements, mc * sizeof(uint16_t), in f16_spmm()