Lines Matching refs:kernel_n
29 std::uint8_t* PrepareGemmTask(const Params& params, int kernel_m, int kernel_n, 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()
104 template <typename Executor, typename Params, int kernel_m, int kernel_n,
110 Gemm<Executor, Params, kernel_m, kernel_n, kernel_k>(params); in Run()
119 int kernel_m, int kernel_n, int kernel_k>
122 typedef internal::GemmTaskRunner<Executor, Params, kernel_m, kernel_n, 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()