pytorch/caffe2/python/examples
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
..
__init__.py Add __init__.py so files get picked up on install (#14898) 2018-12-07 13:40:23 -08:00
char_rnn.py
lmdb_create_example.py
resnet50_trainer.py Enhance cpu support on gloo based multi-nodes mode. (#11330) 2019-01-15 11:47:10 -08:00