Lines Matching refs:group_channels
36 inline ChannelShuffleOperatorTester& group_channels(size_t group_channels) { in group_channels() function
37 assert(group_channels != 0); in group_channels()
38 this->group_channels_ = group_channels; in group_channels()
42 inline size_t group_channels() const { in group_channels() function
47 return groups() * group_channels(); in channels()
116 groups(), group_channels(), in TestX8()
137 for (size_t c = 0; c < group_channels(); c++) { in TestX8()
138 ASSERT_EQ(uint32_t(input[i * input_stride() + g * group_channels() + c]), in TestX8()
164 groups(), group_channels(), in TestX32()
185 for (size_t c = 0; c < group_channels(); c++) { in TestX32()
186 ASSERT_EQ(input[i * input_stride() + g * group_channels() + c], in TestX32()