24   const size_t group_channels = static_cast<size_t>(state.range(2));  in channel_shuffle_x8()  local
30 …d::vector<uint8_t> input(XNN_EXTRA_BYTES / sizeof(uint8_t) + batch_size * groups * group_channels);  in channel_shuffle_x8()
31   std::vector<uint8_t> output(batch_size * groups * group_channels);  in channel_shuffle_x8()
42     groups, group_channels,  in channel_shuffle_x8()
43     groups * group_channels /* input stride */,  in channel_shuffle_x8()
44     groups * group_channels /* output stride */,  in channel_shuffle_x8()
77   const size_t elements_per_iteration = batch_size * groups * group_channels;  in channel_shuffle_x8()
89   const size_t group_channels = static_cast<size_t>(state.range(2));  in channel_shuffle_x32()  local
95   std::vector<float> input(XNN_EXTRA_BYTES / sizeof(float) + batch_size * groups * group_channels);  in channel_shuffle_x32()
96   std::vector<float> output(batch_size * groups * group_channels);  in channel_shuffle_x32()
107     groups, group_channels,  in channel_shuffle_x32()
108     groups * group_channels /* input stride */,  in channel_shuffle_x32()
109     groups * group_channels /* output stride */,  in channel_shuffle_x32()
142   const size_t elements_per_iteration = batch_size * groups * group_channels;  in channel_shuffle_x32()