Lines Matching refs:lanes_count
64 static int Scratch(const RowMajor& params, int lanes_count, int pack_size) { in Scratch() argument
65 return AlignTo<64>(lanes_count * AlignTo(pack_size, params.stride)); in Scratch()
87 static int Scratch(const RowMajorWithSum& params, int lanes_count, in Scratch() argument
89 return 32 + AlignTo<32>(sizeof(InType) * lanes_count * in Scratch()
112 static int Scratch(const ColumnMajorWithSum& params, int lanes_count, in Scratch() argument
114 return 32 + AlignTo<32>(sizeof(InType) * lanes_count * in Scratch()
119 template <typename InType, int lanes_count, int pack_size, int leftovers>
120 class Stream<InType, lanes_count, pack_size, leftovers, RowMajor> {
126 << ")::Pack() -- " << lanes_count << "x" << pack_size << " + " in Pack()
130 if (lanes_count != 0) { in Pack()
142 return sizeof(InType) * pack_size * lanes_count; in PackedAdvance()
146 return lanes_count * params.stride; in UnpackedStride()
150 return AlignTo<32>(lanes_count * AlignTo<pack_size>(params.stride)); in PackedStride()
159 std::cout << " dims: " << lanes_count << "x" << pack_size << " + " in Debug()
174 template <typename InType, int lanes_count, int pack_size, int leftovers>
175 class Stream<InType, lanes_count, pack_size, leftovers, RowMajorWithSum> {
182 << lanes_count << "x" << pack_size << " + " << leftovers in Pack()
186 if (lanes_count != 0) { in Pack()
198 return sizeof(InType) * pack_size * lanes_count; in PackedAdvance()
202 return sizeof(InType) * lanes_count * params.stride; in UnpackedStride()
206 return 32 + AlignTo<32>(sizeof(InType) * lanes_count * in PackedStride()
219 std::cout << " dims: " << lanes_count << "x" << pack_size << " + " in Debug()
238 template <typename InType, int lanes_count, int pack_size, int leftovers>
239 class Stream<InType, lanes_count, pack_size, leftovers, ColumnMajorWithSum> {
246 << ")::Pack() -- " << lanes_count << "x" << pack_size << " + " in Pack()
250 if (lanes_count != 0) { in Pack()
263 return sizeof(InType) * pack_size * lanes_count; in PackedAdvance()
267 return sizeof(InType) * lanes_count; in UnpackedStride()
271 return 32 + AlignTo<32>(sizeof(InType) * lanes_count * in PackedStride()
284 std::cout << " dims: " << lanes_count << "x" << pack_size << " + " in Debug()