Lines Matching refs:scratch
34 void CacheFriendlyMatrixMatrix(std::uint8_t* scratch, const IN_TYPE* lhs,
46 scratch, lhs, rhs + i * chunk_size, m, optimal_n, k,
51 scratch, lhs, rhs + chunks_count_less_one * chunk_size, m, n_left, k,
54 operation.ExecuteCacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k,
67 void ExecuteMatrixMatrix(std::uint8_t* scratch, const std::uint8_t* lhs,
71 CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, result_stride,
75 void ExecuteCacheFriendlyMatrixMatrix(std::uint8_t* scratch,
81 gemm_q8_strided(scratch, lhs, rhs, m, n, k, lhs_offset, rhs_offset,
97 void ExecuteMatrixMatrix(std::uint8_t* scratch, const std::uint8_t* lhs,
101 CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, result_stride,
105 void ExecuteCacheFriendlyMatrixMatrix(std::uint8_t* scratch,
111 gemm_f_strided(scratch, lhs, rhs, m, n, k, lhs_offset, rhs_offset,
126 void ExecuteMatrixMatrix(std::uint8_t* scratch, const std::uint8_t* lhs,
130 CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, result_stride,
134 void ExecuteCacheFriendlyMatrixMatrix(std::uint8_t* scratch,
140 gemm_i32_strided(scratch, lhs, rhs, m, n, k, lhs_offset, rhs_offset, result,
159 std::uint8_t* scratch, const std::uint8_t* lhs,
166 multi_thread_gemv_q8(pool, max_threads, scratch, lhs, rhs, n, k, lhs_offset,
170 multi_thread_gemv_q8(pool, max_threads, scratch, rhs, lhs, m, k, rhs_offset,
179 internal::CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, n,
182 internal::MultiThreadedMatrixMatrix(pool, max_threads, scratch, lhs, rhs, m,
194 std::uint8_t* scratch, const std::uint8_t* lhs,
200 multi_thread_gemv_f(pool, max_threads, scratch, lhs, rhs, n, k, lhs_offset,
204 multi_thread_gemv_f(pool, max_threads, scratch, rhs, lhs, m, k, rhs_offset,
212 internal::CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, n,
215 internal::MultiThreadedMatrixMatrix(pool, max_threads, scratch, lhs, rhs, m,
227 std::int32_t max_threads, std::uint8_t* scratch,
233 multi_thread_gemv_i32(pool, max_threads, scratch, lhs, rhs, n, k,
237 multi_thread_gemv_i32(pool, max_threads, scratch, rhs, lhs, m, k,
245 internal::CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, n,
248 internal::MultiThreadedMatrixMatrix(pool, max_threads, scratch, lhs, rhs, m,
