113   inline ArgMaxPoolMicrokernelTester& channels(size_t channels) {  in channels()  argument
114     assert(channels != 0);  in channels()
115     this->channels_ = channels;  in channels()
119   inline size_t channels() const {  in channels()  function
131       return channels();  in output_stride()
133       assert(this->output_stride_ >= channels());  in output_stride()
172       ((output_pixels() - 1) * step() + pooling_elements()) * channels());
173     std::vector<float> output((output_pixels() - 1) * output_stride() + channels());
174     std::vector<uint32_t> index(output_pixels() * channels());
175     std::vector<float> output_ref(output_pixels() * channels());
176     std::vector<uint32_t> index_ref(output_pixels() * channels());
182         indirect_input[i] = input.data() + i * channels() - input_offset();
189         for (size_t c = 0; c < channels(); c++) {
199           output_ref[x * channels() + c] = max_value;
200           index_ref[x * channels() + c] = max_index;
228       argmaxpool(output_pixels(), pooling_elements(), channels(),
231         (output_stride() - channels()) * sizeof(float),
236         for (size_t c = 0; c < channels(); c++) {
238 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
242 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
245           ASSERT_EQ(output_ref[x * channels() + c], output[x * output_stride() + c])
246 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
250               indirect_input[x * step() + index_ref[x * channels() + c]][c + input_offset()],
251               indirect_input[x * step() + index[x * channels() + c]][c + input_offset()])
252 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
255           ASSERT_EQ(index_ref[x * channels() + c], index[x * channels() + c])
256 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
271       ((output_pixels() - 1) * step() + pooling_elements()) * channels());
272     std::vector<float> output((output_pixels() - 1) * output_stride() + channels());
273     std::vector<uint32_t> index(output_pixels() * channels());
275       channels() + XNN_EXTRA_BYTES / sizeof(uint32_t));
277       channels() + XNN_EXTRA_BYTES / sizeof(float));
278     std::vector<float> output_ref(output_pixels() * channels());
279     std::vector<uint32_t> index_ref(output_pixels() * channels());
285         indirect_input[i] = input.data() + i * channels() - input_offset();
292         for (size_t c = 0; c < channels(); c++) {
302           output_ref[x * channels() + c] = max_value;
303           index_ref[x * channels() + c] = max_index;
331       argmaxpool(output_pixels(), pooling_elements(), channels(),
336         (output_stride() - channels()) * sizeof(float),
341         for (size_t c = 0; c < channels(); c++) {
343 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
347 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
350           ASSERT_EQ(output_ref[x * channels() + c], output[x * output_stride() + c])
351 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
355               indirect_input[x * step() + index_ref[x * channels() + c]][c + input_offset()],
356               indirect_input[x * step() + index[x * channels() + c]][c + input_offset()])
357 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
360           ASSERT_EQ(index_ref[x * channels() + c], index[x * channels() + c])
361 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()