• Home
  • Raw
  • Download

Lines Matching refs:aligned

62 def GenerateZipVector(emitter, aligned, leftovers):  argument
64 zip_Nx8_neon.BuildName(1, leftovers, aligned),
83 def GenerateMulCols(emitter, result_type, lhs_add, rhs_add, aligned, cols, argument
88 zip_Nx8_neon.BuildName(4, leftovers, aligned),
93 zip_Nx8_neon.BuildName(4, leftovers, aligned),
106 zip_Nx8_neon.BuildName(4, leftovers, aligned),
111 zip_Nx8_neon.BuildName(cols - 4, leftovers, aligned),
119 zip_Nx8_neon.BuildName(cols, leftovers, aligned),
127 def GenerateQuantized8BitMul(emitter, aligned, cols, leftovers): argument
129 GenerateMulCols(emitter, 'int32', False, True, aligned, cols, leftovers)
131 qnt_Nx8_neon.BuildName(1, cols, aligned),
136 def GenerateFullMul(emitter, result_type, aligned, cols, leftovers): argument
137 GenerateMulCols(emitter, result_type, True, True, aligned, cols, leftovers)
140 def BuildName(output_type, aligned, cols, leftover): argument
142 if aligned:
170 def GenerateGemv(emitter, output_type, aligned, cols, leftovers): argument
173 BuildName(output_type, aligned, cols, leftovers),
181 GenerateZipVector(emitter, aligned, leftovers)
182 GenerateQuantized8BitMul(emitter, aligned, cols, leftovers)
185 GenerateZipVector(emitter, aligned, leftovers)
186 GenerateFullMul(emitter, 'int32', aligned, cols, leftovers)
189 GenerateZipVector(emitter, aligned, leftovers)
190 GenerateFullMul(emitter, 'float', aligned, cols, leftovers)
197 def GenerateGemvCall(emitter, output_type, aligned, m_mod, leftovers): argument
200 BuildName(output_type, aligned, m_mod, leftovers)),
204 def GenerateGemvSwitch2(emitter, output_type, aligned, n_mod): argument
211 GenerateGemvCall(emitter, output_type, aligned, n_mod, leftovers)
218 def GenerateGemvSwitch1(emitter, output_type, aligned): argument
225 GenerateGemvSwitch2(emitter, output_type, aligned, n_mod)
275 for aligned in [True, False]:
278 GenerateGemv(emitter, output_type, aligned, cols, leftover)