Lines Matching refs:group_channels
25 const size_t group_channels = static_cast<size_t>(state.range(2)); in channel_shuffle_x8() local
31 …d::vector<uint8_t> input(XNN_EXTRA_BYTES / sizeof(uint8_t) + batch_size * groups * group_channels); in channel_shuffle_x8()
32 std::vector<uint8_t> output(batch_size * groups * group_channels); in channel_shuffle_x8()
43 groups, group_channels, in channel_shuffle_x8()
44 groups * group_channels /* input stride */, in channel_shuffle_x8()
45 groups * group_channels /* output stride */, in channel_shuffle_x8()
81 const size_t elements_per_iteration = batch_size * groups * group_channels; in channel_shuffle_x8()
93 const size_t group_channels = static_cast<size_t>(state.range(2)); in channel_shuffle_x32() local
99 std::vector<float> input(XNN_EXTRA_BYTES / sizeof(float) + batch_size * groups * group_channels); in channel_shuffle_x32()
100 std::vector<float> output(batch_size * groups * group_channels); in channel_shuffle_x32()
111 groups, group_channels, in channel_shuffle_x32()
112 groups * group_channels /* input stride */, in channel_shuffle_x32()
113 groups * group_channels /* output stride */, in channel_shuffle_x32()
149 const size_t elements_per_iteration = batch_size * groups * group_channels; in channel_shuffle_x32()