Searched refs:kernel_k (Results 1 – 2 of 2) sorted by relevance
| /external/gemmlowp/meta/ |
| D | multi_thread_gemm.h | 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>
[all …]
|
| D | single_thread_gemm.h | 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>
[all …]
|