Lines Matching refs:lanes
26 def BuildName(lanes, leftovers, aligned): argument
27 name = 'qnt_%dx8' % lanes
35 def LoadAndDuplicateOffsets(emitter, registers, lanes, offsets): argument
36 if lanes == 1 or lanes == 2 or lanes == 3:
38 for unused_i in range(0, lanes):
46 raise ConfigurationError('Unsupported number of lanes: %d' % lanes)
55 lanes = []
60 lanes.append(QntLane(source,
68 lanes.append(QntLane(input_register,
77 return lanes
86 def GenerateQuantize(emitter, registers, lanes, lane_temps, argument
89 for lane in lanes:
92 for lane in lanes:
95 for lane in lanes:
98 for lane in lanes:
101 for lane in lanes:
108 def GenerateLoadQuantizeStore(emitter, registers, lanes, multiplicative_offset, argument
112 for lane in lanes:
115 for lane in lanes:
121 for lane in lanes:
125 for (lane_temp, lane) in zip(lane_temps, lanes):
132 for (lane_temp, lane) in zip(lane_temps, lanes):
140 def GenerateLoadLeftovers(emitter, registers, leftovers, lanes): argument
143 for lane in lanes:
148 for lane in lanes:
152 for lane in lanes:
155 for lane in lanes:
160 for lane in lanes:
165 for lane in lanes:
169 for lane in lanes:
174 for lane in lanes:
180 for lane in lanes:
185 for lane in lanes:
193 def GenerateStoreLeftovers(emitter, registers, leftovers, lane_temps, lanes): argument
196 for (temp, lane) in zip(lane_temps, lanes):
246 def GenerateLeftoverLoadQuantizeStore(emitter, registers, leftovers, lanes, argument
251 for lane in lanes:
254 GenerateLoadLeftovers(emitter, registers, leftovers, lanes)
257 for (lane_temp, lane) in zip(lane_temps, lanes):
266 GenerateStoreLeftovers(emitter, registers, leftovers, lane_temps, lanes)
304 lanes = GenerateQntLanes(
318 GenerateLoadQuantizeStore(emitter, registers, lanes, multiplicative_offset,
326 GenerateLeftoverLoadQuantizeStore(emitter, registers, leftovers, lanes,
371 for lanes in range(1, 4):
373 GenerateQntNx8(neon, lanes, leftovers, aligned)