Searched refs:src_batch_bytesize (Results 1 – 1 of 1) sorted by relevance
1262 auto src_batch_bytesize = self_sizes[1] * block_bytesize; in index_select_out_cpu_dim1_() local1277 (const float*)(src_base + batch * src_batch_bytesize); in index_select_out_cpu_dim1_()1291 auto src = src_base + batch * src_batch_bytesize + idx * block_bytesize; in index_select_out_cpu_dim1_()