projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 8a5ba57) | patch
Add NHWC support to Resize Operator (#15553)
authorDavid Carrillo Cisneros <davidca@fb.com>
Wed, 9 Jan 2019 00:04:41 +0000 (16:04 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 9 Jan 2019 00:44:17 +0000 (16:44 -0800)
commit2b226122899c4fdb66432aa740044fcca9d25b2d
treeff8c209e523cc9e5bf144a561be8808c735cbcaatree | snapshot
parent8a5ba577c17d039b07270d66a9607c770fbcf499commit | diff
Add NHWC support to Resize Operator (#15553)

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

Add unit test and implementation of NHWC layout for Resize operator.

Also, add pragma parallel loop to old NCHWC layout.

Reviewed By: jspark1105

Differential Revision: D13540762

fbshipit-source-id: eebf252bf0d1efdff180a171d804181045f100a5
caffe2/operators/resize_op.cc diff | blob | history
caffe2/operators/resize_op.h diff | blob | history
caffe2/python/operator_test/resize_op_test.py diff | blob | history
Domain: Machine Learning / ML Framework;