Lines Matching refs:leftovers
76 def ZipName(rows, leftovers, aligned): argument
77 return zip_Nx8_neon.BuildName(rows, leftovers, aligned)
80 def GenerateZipRhs(emitter, aligned, cols, leftovers): argument
84 ZipName(3, leftovers, aligned),
92 ZipName(cols, leftovers, aligned),
110 rows, cols, leftovers): argument
113 ZipName(rows, leftovers, aligned),
134 def GenerateQuantized8BitMul(emitter, aligned, rows, cols, leftovers): argument
138 cols, leftovers)
151 cols, leftovers)
159 def GenerateFullMul(emitter, result_type, aligned, rows, cols, leftovers): argument
162 cols, leftovers)
170 rows, cols, leftovers)
209 def GenerateGemm(emitter, output_type, aligned, rows, cols, leftovers): argument
212 BuildName(output_type, aligned, rows, cols, leftovers),
217 emitter.EmitAssert('k %% 8 == %d' % leftovers)
221 GenerateZipRhs(emitter, aligned, cols, leftovers)
222 GenerateQuantized8BitMul(emitter, aligned, rows, cols, leftovers)
225 GenerateZipRhs(emitter, aligned, cols, leftovers)
226 GenerateFullMul(emitter, 'int32', aligned, rows, cols, leftovers)
229 GenerateZipRhs(emitter, aligned, cols, leftovers)
230 GenerateFullMul(emitter, 'float', aligned, rows, cols, leftovers)
237 def GenerateGemmCall(emitter, output_type, aligned, m_mod, n_mod, leftovers): argument
240 BuildName(output_type, aligned, m_mod, n_mod, leftovers)),
248 for leftovers in range(0, 8):
249 emitter.EmitCase(leftovers)
251 GenerateGemmCall(emitter, output_type, aligned, m_mod, n_mod, leftovers)