25           int kernel_k>
32                                  int kernel_k) {  in EstimateScratchSize()  argument
35             params.left_stream, kernel_m, kernel_k);  in EstimateScratchSize()
40             params.right_stream, kernel_n, kernel_k);  in EstimateScratchSize()
221                                  int kernel_k) {  in EstimateScratchSize()  argument
226             params.left_stream, kernel_m, kernel_k);  in EstimateScratchSize()
229             params.right_stream, kernel_n, kernel_k);  in EstimateScratchSize()
442                                  int kernel_k) {  in EstimateScratchSize()  argument
491                                  int kernel_k) {  in EstimateScratchSize()  argument
677           int kernel_k>
679   internal::Dispatch3DStage1<Executor, Params, kernel_m, kernel_n, kernel_k,
682                                                     params.k % kernel_k);