Home
last modified time | relevance | path

Searched defs:BLOCK_N (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
Dint4mm_kernel.cpp615 constexpr int BLOCK_N = vec::Vectorized<float>::size() * 4; in weight_to_int4pack_kernel() local
715 constexpr int BLOCK_N = vec::Vectorized<float>::size() * 4; in int4pack_mm_kernel_() local
Dint8mm_kernel.cpp375 constexpr int BLOCK_N = 4; in int8pack_mm_kernel_() local