Created by: NHZlX
ceil mode 下,TRT对 avg pool 支持不好。因此使用plugin的方式实现 avg pool, 同时复用 fluid pool op kernel.
eg:
shape 大小为1 * 3 * 3输入,kernel 是2 * 2,stride为(2, 2), padding (0, 0)
,floor mode时候 输出为1 * 1 * 1, ceil mode 下输出为1 * 2 * 2。
ceil mode下需要对输入加padding,并且保证padding值不加入平均值计算中。