Lines Matching refs:kernel_m
24 template <typename Executor, typename Params, int kernel_m, int kernel_n,
31 static int EstimateScratchSize(const P& params, int kernel_m, int kernel_n, in EstimateScratchSize() argument
35 params.left_stream, kernel_m, kernel_k); in EstimateScratchSize()
220 static int EstimateScratchSize(const P& params, int kernel_m, int kernel_n, in EstimateScratchSize() argument
222 const int lhs_chunks = ((params.m + kernel_m - 1) / kernel_m); in EstimateScratchSize()
226 params.left_stream, kernel_m, kernel_k); in EstimateScratchSize()
441 static int EstimateScratchSize(const P& params, int kernel_m, int kernel_n, in EstimateScratchSize() argument
490 static int EstimateScratchSize(const P& params, int kernel_m, int kernel_n, in EstimateScratchSize() argument
676 template <typename Executor, typename Params, int kernel_m, int kernel_n,
679 internal::Dispatch3DStage1<Executor, Params, kernel_m, kernel_n, kernel_k,
680 kernel_m - 1>::Execute(params, params.m % kernel_m,