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
36     const int rhs_chunks = ((params.n + kernel_n - 1) / kernel_n);  in EstimateScratchSize()
40             params.right_stream, kernel_n, kernel_k);  in EstimateScratchSize()
220   static int EstimateScratchSize(const P& params, int kernel_m, int kernel_n,  in EstimateScratchSize()  argument
229             params.right_stream, kernel_n, 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,
681                                                     params.n % kernel_n,