Lines Matching refs:kernel_k
30 int kernel_k, std::uint8_t* scratch, int m_start, in PrepareGemmTask() argument
53 task, kernel_m, kernel_n, kernel_k); in PrepareGemmTask()
58 int kernel_m, int kernel_n, int kernel_k, in PrepareGemmTasks() argument
81 params, kernel_m, kernel_n, kernel_k, scratch, i * m_chunk, m_chunk, in PrepareGemmTasks()
85 PrepareGemmTask<Executor, Params>(params, kernel_m, kernel_n, kernel_k, in PrepareGemmTasks()
92 params, kernel_m, kernel_n, kernel_k, scratch, 0, params.m, in PrepareGemmTasks()
96 PrepareGemmTask<Executor, Params>(params, kernel_m, kernel_n, kernel_k, in PrepareGemmTasks()
105 int kernel_k>
110 Gemm<Executor, Params, kernel_m, kernel_n, kernel_k>(params); in Run()
119 int kernel_m, int kernel_n, int kernel_k>
123 kernel_k> in MultiThreadGemm()
128 context, params, kernel_m, kernel_n, kernel_k, &task_params)) { in MultiThreadGemm()
129 Gemm<Executor, Params, kernel_m, kernel_n, kernel_k>(params); in MultiThreadGemm()