Lines Matching refs:group_channels
37 inline ChannelShuffleOperatorTester& group_channels(size_t group_channels) { in group_channels() argument
38 assert(group_channels != 0); in group_channels()
39 this->group_channels_ = group_channels; in group_channels()
43 inline size_t group_channels() const { in group_channels() function
48 return groups() * group_channels(); in channels()
117 groups(), group_channels(), in TestX8()
138 for (size_t c = 0; c < group_channels(); c++) { in TestX8()
139 ASSERT_EQ(uint32_t(input[i * input_stride() + g * group_channels() + c]), in TestX8()
165 groups(), group_channels(), in TestX32()
186 for (size_t c = 0; c < group_channels(); c++) { in TestX32()
187 ASSERT_EQ(input[i * input_stride() + g * group_channels() + c], in TestX32()