add channel last support for MaxUnpool2d (#49984)
authormingfeima <mingfei.ma@intel.com>
Mon, 30 Aug 2021 01:35:37 +0000 (18:35 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Mon, 30 Aug 2021 01:37:10 +0000 (18:37 -0700)
commitc5ed31e4a7550bfe5a4893b3803ee7fdf1b31f53
tree5a6cf508d16d8e3ddd2813b0b36060876739fd78
parent9db56531f708188cf59d9d4db60871405df9df69
add channel last support for MaxUnpool2d (#49984)

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

Test Plan: Imported from OSS

Reviewed By: ejguan

Differential Revision: D26007051

Pulled By: VitalyFedyunin

fbshipit-source-id: 6c54751ade4092e03c1651aaa60380f7d6e92f6b
aten/src/ATen/native/MaxUnpooling.cpp
aten/src/ATen/native/cpu/MaxUnpoolKernel.cpp [new file with mode: 0644]
aten/src/ATen/native/cpu/MaxUnpoolKernel.h [new file with mode: 0644]
test/test_nn.py
tools/build_variables.bzl