Lines Matching refs:output_pixels

31   inline ArgMaxPoolMicrokernelTester& output_pixels(size_t output_pixels) {  in output_pixels()  argument
32     assert(output_pixels != 0);  in output_pixels()
33     this->output_pixels_ = output_pixels;  in output_pixels()
37   inline size_t output_pixels() const {  in output_pixels()  function
170 …std::vector<const float*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elements(…
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());
181       for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) {
185         indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
188       for (size_t x = 0; x < output_pixels(); x++) {
228       argmaxpool(output_pixels(), pooling_elements(), channels(),
235       for (size_t x = 0; x < output_pixels(); x++) {
238 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
242 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
246 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
252 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
256 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
269 …std::vector<const float*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elements(…
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());
278     std::vector<float> output_ref(output_pixels() * channels());
279     std::vector<uint32_t> index_ref(output_pixels() * channels());
284       for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) {
288         indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
291       for (size_t x = 0; x < output_pixels(); x++) {
331       argmaxpool(output_pixels(), pooling_elements(), channels(),
340       for (size_t x = 0; x < output_pixels(); x++) {
343 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
347 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
351 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
357 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()
361 …        << "at pixel " << x << " / " << output_pixels() << ", channel " << c << " / " << channels()