Lines Matching refs:pack_size
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()
88 int pack_size) { in Scratch() argument
90 AlignTo(pack_size, params.count)); in Scratch()
113 int pack_size) { in Scratch() argument
115 AlignTo(pack_size, params.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()
138 return sizeof(InType) * pack_size; in UnpackedAdvance()
142 return sizeof(InType) * pack_size * lanes_count; in PackedAdvance()
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()
194 return sizeof(InType) * pack_size; in UnpackedAdvance()
198 return sizeof(InType) * pack_size * lanes_count; in PackedAdvance()
207 AlignTo<pack_size>(params.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()
259 return sizeof(InType) * pack_size * params.stride; in UnpackedAdvance()
263 return sizeof(InType) * pack_size * lanes_count; in PackedAdvance()
272 AlignTo<pack_size>(params.count)); in PackedStride()
284 std::cout << " dims: " << lanes_count << "x" << pack_size << " + " in Debug()