D | gemm_NxMxK_neon.py | 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), 109 def GenerateMulRows(emitter, result, result_type, lhs_add, rhs_add, aligned, argument 113 ZipName(rows, leftovers, aligned), 134 def GenerateQuantized8BitMul(emitter, aligned, rows, cols, leftovers): argument 137 GenerateMulRows(emitter, 'temp_result', 'int32', False, True, aligned, 3, 140 qnt_Nx8_neon.BuildMultiQuantizeName(aligned, 3), [all …]
|