Lines Matching refs:cols
101 def GenerateZipRhs(emitter, aligned, cols, leftovers):
111 if cols is not 0:
113 ZipName(cols, leftovers, aligned),
118 def MulName(result_type, lhs_add, rhs_add, rows, cols):
119 return mul_Nx8_Mx8_neon.BuildName(result_type, lhs_add, rhs_add, rows, cols)
131 rows, cols, leftovers):
149 if cols is not 0:
151 MulName(result_type, lhs_add, rhs_add, rows, cols),
155 def GenerateQuantized8BitMul(emitter, aligned, rows, cols, leftovers):
159 cols, leftovers)
172 cols, leftovers)
180 def GenerateFullMul(emitter, result_type, aligned, rows, cols, leftovers):
183 cols, leftovers)
191 rows, cols, leftovers)
194 def BuildName(output_type, aligned, rows, cols, leftover):
195 name = BuildMainGemmName(output_type) + '_%d_%d_%d' % (rows, cols, leftover)
230 def GenerateGemm(emitter, output_type, aligned, rows, cols, leftovers):
233 BuildName(output_type, aligned, rows, cols, leftovers),
237 emitter.EmitAssert('n %% 3 == %d' % cols)
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)
314 """Second level of main switch, choose optimized version on cols leftover."""
425 for cols in range(0, 3):
427 GenerateGemm(emitter, output_type, aligned, rows, cols, leftover)
