Pull Request resolved: https://github.com/pytorch/pytorch/pull/182438
Approved by: https://github.com/chuanqi129, https://github.com/jansel

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