Searched defs:kernel_k (Results 1 – 2 of 2) sorted by relevance
30 int kernel_k, std::uint8_t* scratch, int m_start, in PrepareGemmTask()58 int kernel_m, int kernel_n, int kernel_k, in PrepareGemmTasks()
32 int kernel_k) { in EstimateScratchSize()221 int kernel_k) { in EstimateScratchSize()442 int kernel_k) { in EstimateScratchSize()491 int kernel_k) { in EstimateScratchSize()