Shane Li 620ff25bdb Enhance cpu support on gloo based multi-nodes mode. (#11330) ... Summary: 1. Add some gloo communication operators into related fallback list; 2. Work around to avoid compiling errors while using fallback operator whose CPU operator inherits from 'OperatorBase' directly like PrefetchOperator; 3. Add new cpu context support for some python module files and resnet50 training example file. Pull Request resolved: https://github.com/pytorch/pytorch/pull/11330 Reviewed By: yinghai Differential Revision: D13624519 Pulled By: wesolwsk fbshipit-source-id: ce39d57ddb8cd7786db2e873bfe954069d972f4f		2019-01-15 11:47:10 -08:00
..
CMakeLists.txt	Change hip filename extension to .hip (#14036)	2018-11-16 11:55:59 -08:00
image_input_op.cc	Enhance cpu support on gloo based multi-nodes mode. (#11330)	2019-01-15 11:47:10 -08:00
image_input_op.h	Tensor reinitialization codemod - 2/5 (#15947)	2019-01-11 15:05:01 -08:00
image_input_op_gpu.cc	Windows shared build (#13550)	2018-11-16 12:16:28 -08:00
transform_gpu.cu	Remove partially initialized Tensor + CopyFrom (#13629)	2018-11-07 11:31:03 -08:00
transform_gpu.h