3 // This source code is licensed under the BSD-style license found in the
25     this->padding_top_ = padding;  in padding()
26     this->padding_right_ = padding;  in padding()
27     this->padding_bottom_ = padding;  in padding()
28     this->padding_left_ = padding;  in padding()
33     this->padding_top_ = padding_height;  in padding()
34     this->padding_right_ = padding_width;  in padding()
35     this->padding_bottom_ = padding_height;  in padding()
36     this->padding_left_ = padding_width;  in padding()
41     this->padding_top_ = padding_height;  in padding_height()
42     this->padding_bottom_ = padding_height;  in padding_height()
47     this->padding_right_ = padding_width;  in padding_width()
48     this->padding_left_ = padding_width;  in padding_width()
53     this->padding_top_ = padding_top;  in padding_top()
58     return this->padding_top_;  in padding_top()
62     this->padding_right_ = padding_right;  in padding_right()
67     return this->padding_right_;  in padding_right()
71     this->padding_bottom_ = padding_bottom;  in padding_bottom()
76     return this->padding_bottom_;  in padding_bottom()
80     this->padding_left_ = padding_left;  in padding_left()
85     return this->padding_left_;  in padding_left()
91     this->input_height_ = input_height;  in input_size()
92     this->input_width_ = input_width;  in input_size()
98     this->input_height_ = input_height;  in input_height()
103     return this->input_height_;  in input_height()
108     this->input_width_ = input_width;  in input_width()
113     return this->input_width_;  in input_width()
118     this->channels_ = channels;  in channels()
123     return this->channels_;  in channels()
128     this->batch_size_ = batch_size;  in batch_size()
133     return this->batch_size_;  in batch_size()
138     this->pooling_height_ = pooling_size;  in pooling_size()
139     this->pooling_width_ = pooling_size;  in pooling_size()
146     this->pooling_height_ = pooling_height;  in pooling_size()
147     this->pooling_width_ = pooling_width;  in pooling_size()
153     this->pooling_height_ = pooling_height;  in pooling_height()
158     return this->pooling_height_;  in pooling_height()
163     this->pooling_width_ = pooling_width;  in pooling_width()
168     return this->pooling_width_;  in pooling_width()
173     return std::max<size_t>(input_height() * pooling_height(), padding_height) - padding_height;  in output_height()
178     return std::max<size_t>(input_width() * pooling_width(), padding_width) - padding_width;  in output_width()
183     this->input_pixel_stride_ = input_pixel_stride;  in input_pixel_stride()
188     if (this->input_pixel_stride_ == 0) {  in input_pixel_stride()
191       assert(this->input_pixel_stride_ >= channels());  in input_pixel_stride()
192       return this->input_pixel_stride_;  in input_pixel_stride()
198     this->output_pixel_stride_ = output_pixel_stride;  in output_pixel_stride()
203     if (this->output_pixel_stride_ == 0) {  in output_pixel_stride()
206       assert(this->output_pixel_stride_ >= channels());  in output_pixel_stride()
207       return this->output_pixel_stride_;  in output_pixel_stride()
214     this->next_input_height_ = next_input_height;  in next_input_size()
215     this->next_input_width_ = next_input_width;  in next_input_size()
221     this->next_input_height_ = next_input_height;  in next_input_height()
226     if (this->next_input_height_ == 0) {  in next_input_height()
229       return this->next_input_height_;  in next_input_height()
235     this->next_input_width_ = next_input_width;  in next_input_width()
240     if (this->next_input_width_ == 0) {  in next_input_width()
243       return this->next_input_width_;  in next_input_width()
249 …  return std::max<size_t>(next_input_height() * pooling_height(), padding_height) - padding_height;  in next_output_height()
254     return std::max<size_t>(next_input_width() * pooling_width(), padding_width) - padding_width;  in next_output_width()
259     this->next_batch_size_ = next_batch_size;  in next_batch_size()
264     if (this->next_batch_size_ == 0) {  in next_batch_size()
267       return this->next_batch_size_;  in next_batch_size()
272     this->iterations_ = iterations;  in iterations()
277     return this->iterations_;  in iterations()
284 …std::uniform_int_distribution<uint32_t>(0, pooling_height() * pooling_width() - 1), std::ref(rng));  in TestX32()
286 …std::vector<uint32_t> input((batch_size() * input_height() * input_width() - 1) * input_pixel_stri…  in TestX32()
287     std::vector<uint32_t> index(batch_size() * input_height() * input_width() * channels());  in TestX32()
288 …std::vector<uint32_t> output((batch_size() * output_height() * output_width() - 1) * output_pixel_…  in TestX32()
291       std::generate(input.begin(), input.end(), std::ref(u32rng));  in TestX32()
292       std::generate(index.begin(), index.end(), std::ref(idx_rng));  in TestX32()
293       std::generate(output.begin(), output.end(), std::ref(u32rng));  in TestX32()
301 …const uint32_t pooling_index = index[((i * input_height() + iy) * input_width() + ix) * channels()…  in TestX32()
304 …(std::max<size_t>(iy * pooling_height() + py, padding_top()) - padding_top(), output_height() - 1);  in TestX32()
305 …(std::max<size_t>(ix * pooling_width() + px, padding_left()) - padding_left(), output_width() - 1);  in TestX32()
332           input.data(), index.data(), output.data(),  in TestX32()
345                 "in batch index " << i << ", pixel (" << y << ", " << x << "), channel " << c;  in TestX32()
357 …std::uniform_int_distribution<uint32_t>(0, pooling_height() * pooling_width() - 1), std::ref(rng));  in TestSetupX32()
360       (batch_size() * input_height() * input_width() - 1) * input_pixel_stride() + channels(),  in TestSetupX32()
361 …(next_batch_size() * next_input_height() * next_input_width() - 1) * input_pixel_stride() + channe…  in TestSetupX32()
362     std::vector<uint32_t> index(std::max<size_t>(  in TestSetupX32()
366       (batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channels(),  in TestSetupX32()
367 …(next_batch_size() * next_output_height() * next_output_width() - 1) * output_pixel_stride() * cha…  in TestSetupX32()
372       std::generate(input.begin(), input.end(), std::ref(u32rng));  in TestSetupX32()
373       std::generate(index.begin(), index.end(), std::ref(idx_rng));  in TestSetupX32()
374       std::generate(output.begin(), output.end(), std::ref(u32rng));  in TestSetupX32()
382 …const uint32_t pooling_index = index[((i * input_height() + iy) * input_width() + ix) * channels()…  in TestSetupX32()
385 …(std::max<size_t>(iy * pooling_height() + py, padding_top()) - padding_top(), output_height() - 1);  in TestSetupX32()
386 …(std::max<size_t>(ix * pooling_width() + px, padding_left()) - padding_left(), output_width() - 1);  in TestSetupX32()
413           input.data(), index.data(), output.data(),  in TestSetupX32()
426                 "in batch index " << i << ", pixel (" << y << ", " << x << "), channel " << c;  in TestSetupX32()
432       // Re-generate data for the second run.  in TestSetupX32()
433       std::generate(input.begin(), input.end(), std::ref(u32rng));  in TestSetupX32()
434       std::generate(index.begin(), index.end(), std::ref(idx_rng));  in TestSetupX32()
435       std::generate(output.begin(), output.end(), std::ref(u32rng));  in TestSetupX32()
443 …const uint32_t pooling_index = index[((i * next_input_height() + iy) * next_input_width() + ix) * …  in TestSetupX32()
446 …::max<size_t>(iy * pooling_height() + py, padding_top()) - padding_top(), next_output_height() - 1…  in TestSetupX32()
447 …::max<size_t>(ix * pooling_width() + px, padding_left()) - padding_left(), next_output_width() - 1…  in TestSetupX32()
460           input.data(), index.data(), output.data(),  in TestSetupX32()
473                 "in batch index " << i << ", pixel (" << y << ", " << x << "), channel " << c;  in TestSetupX32()