| 1358 | } |
| 1359 | |
| 1360 | void pooling:: |
| 1361 | setup_max_pooling( |
| 1362 | int window_height_, |
| 1363 | int window_width_, |
| 1364 | int stride_y_, |
| 1365 | int stride_x_, |
| 1366 | int padding_y_, |
| 1367 | int padding_x_ |
| 1368 | ) |
| 1369 | { |
| 1370 | setup(window_height_, window_width_, stride_y_, stride_x_, padding_y_, padding_x_, CUDNN_POOLING_MAX); |
| 1371 | do_max_pooling = true; |
| 1372 | } |
| 1373 | |
| 1374 | void pooling:: |
| 1375 | setup_avg_pooling( |