Lines Matching refs:leftovers
97 def ZipName(rows, leftovers, aligned): argument
98 return zip_Nx8_neon.BuildName(rows, leftovers, aligned)
101 def GenerateZipRhs(emitter, aligned, cols, leftovers): argument
105 ZipName(3, leftovers, aligned),
113 ZipName(cols, leftovers, aligned),
131 rows, cols, leftovers): argument
134 ZipName(rows, leftovers, aligned),
155 def GenerateQuantized8BitMul(emitter, aligned, rows, cols, leftovers): argument
159 cols, leftovers)
172 cols, leftovers)
180 def GenerateFullMul(emitter, result_type, aligned, rows, cols, leftovers): argument
183 cols, leftovers)
191 rows, cols, leftovers)
230 def GenerateGemm(emitter, output_type, aligned, rows, cols, leftovers): argument
233 BuildName(output_type, aligned, rows, cols, leftovers),
238 emitter.EmitAssert('k %% 8 == %d' % leftovers)
242 GenerateZipRhs(emitter, aligned, cols, leftovers)
243 GenerateQuantized8BitMul(emitter, aligned, rows, cols, leftovers)
246 GenerateZipRhs(emitter, aligned, cols, leftovers)
247 GenerateFullMul(emitter, 'int32', aligned, rows, cols, leftovers)
250 GenerateZipRhs(emitter, aligned, cols, leftovers)
251 GenerateFullMul(emitter, 'float', aligned, rows, cols, leftovers)
277 for leftovers in range(0, 8):
278 emitter.EmitCase(leftovers)
281 qnt_Nx8_neon.BuildName(rows, leftovers, aligned),
292 def GenerateGemmCall(emitter, output_type, aligned, m_mod, n_mod, leftovers): argument
295 BuildName(output_type, aligned, m_mod, n_mod, leftovers)),
303 for leftovers in range(0, 8):
304 emitter.EmitCase(leftovers)
306 GenerateGemmCall(emitter, output_type, aligned, m_mod, n_mod, leftovers)