Pull Request resolved: https://github.com/pytorch/pytorch/pull/188736
Approved by: https://github.com/chuanqi129, https://github.com/etaf, https://github.com/EikanWang, https://github.com/atalman

Co-authored-by: Anatoly Myachev <anatoly.myachev@intel.com>