Adding function to convert Module to channels last
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/28991
Test Plan: Imported from OSS
Differential Revision: D18430810
Pulled By: VitalyFedyunin
fbshipit-source-id: 0693d4e31fc6f9831722c29fc83517f16ddfc028