optimize max pool 2d (#17391)
authorJongsoo Park <jongsoo@fb.com>
Fri, 22 Feb 2019 18:20:24 +0000 (10:20 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 22 Feb 2019 18:36:19 +0000 (10:36 -0800)
commit4778a4089ece23e1ead0e414d1b4d206ce60a192
tree1013352d96e665f711efcd47120c687917ae17bb
parent8a21c6a5eef9b101ce8a56ea2d4a62d5dac85b8d
optimize max pool 2d (#17391)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17391

Optimize 2D max pool using AVX2 intrinsics.

Reviewed By: jianyuh

Differential Revision: D14181620

fbshipit-source-id: ffc6c4412bd1c1d7839fe06226921df40d9cab83
caffe2/quantization/server/pool_dnnlowp_op.cc
caffe2/quantization/server/pool_dnnlowp_op_avx2.cc [new file with mode: 0644]
caffe2/quantization/server/pool_dnnlowp_op_avx2.h [new file with mode: 0644]