Searched defs:nb_size (Results 1 – 2 of 2) sorted by relevance
622 int nb_size = std::min(BLOCK_N, N - i * BLOCK_N); in weight_to_int4pack_kernel() local730 int nb_size = std::min(BLOCK_N, N - nb_start); in int4pack_mm_kernel_() local
390 int nb_size = std::min(BLOCK_N, N - nb_start); in int8pack_mm_kernel_() local