Lines Matching refs:scratch
33 void CacheFriendlyMatrixMatrix(std::uint8_t* scratch, const IN_TYPE* lhs, in CacheFriendlyMatrixMatrix() argument
45 scratch, lhs, rhs + i * chunk_size, m, optimal_n, k, in CacheFriendlyMatrixMatrix()
50 scratch, lhs, rhs + chunks_count_less_one * chunk_size, m, n_left, k, in CacheFriendlyMatrixMatrix()
53 operation.ExecuteCacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, in CacheFriendlyMatrixMatrix()
69 void ExecuteMatrixMatrix(std::uint8_t* scratch, const std::uint8_t* lhs, in ExecuteMatrixMatrix() argument
73 CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, result_stride, in ExecuteMatrixMatrix()
77 void ExecuteCacheFriendlyMatrixMatrix(std::uint8_t* scratch, in ExecuteCacheFriendlyMatrixMatrix() argument
83 gemm_q8_strided(scratch, lhs, rhs, m, n, k, lhs_offset, rhs_offset, in ExecuteCacheFriendlyMatrixMatrix()
108 void ExecuteMatrixMatrix(std::uint8_t* scratch, const std::uint8_t* lhs, in ExecuteMatrixMatrix() argument
112 CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, result_stride, in ExecuteMatrixMatrix()
116 void ExecuteCacheFriendlyMatrixMatrix(std::uint8_t* scratch, in ExecuteCacheFriendlyMatrixMatrix() argument
122 gemm_f_strided(scratch, lhs, rhs, m, n, k, lhs_offset, rhs_offset, in ExecuteCacheFriendlyMatrixMatrix()
142 void ExecuteMatrixMatrix(std::uint8_t* scratch, const std::uint8_t* lhs, in ExecuteMatrixMatrix() argument
146 CacheFriendlyMatrixMatrix(scratch, lhs, rhs, m, n, k, result, result_stride, in ExecuteMatrixMatrix()
150 void ExecuteCacheFriendlyMatrixMatrix(std::uint8_t* scratch, in ExecuteCacheFriendlyMatrixMatrix() argument
156 gemm_i32_strided(scratch, lhs, rhs, m, n, k, lhs_offset, rhs_offset, result, in ExecuteCacheFriendlyMatrixMatrix()
179 std::uint8_t* scratch, const std::uint8_t* lhs, in multi_thread_gemm_q8() argument
187 internal::MultiThreadedMatrixMatrix(pool, max_threads, scratch, lhs, rhs, m, in multi_thread_gemm_q8()
198 std::uint8_t* scratch, const std::uint8_t* lhs, in multi_thread_gemm_f() argument
204 internal::MultiThreadedMatrixMatrix(pool, max_threads, scratch, lhs, rhs, m, in multi_thread_gemm_f()
215 std::int32_t max_threads, std::uint8_t* scratch, in multi_thread_gemm_i32() argument
221 internal::MultiThreadedMatrixMatrix(pool, max_threads, scratch, lhs, rhs, m, in multi_thread_gemm_i32()