saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/python/predictor
History
Davin Wang f2be3a4e5e Allow specifying device to prepare_prediction_net() 
Summary:
This is a supplementary to commit ce8267d425444f60ae650389fb41838847a44a5e. It allows specifying device to prepare_prediction_net() so prediction extractor can work with GPU.
Closes https://github.com/caffe2/caffe2/pull/1035

Differential Revision: D6467420

Pulled By: salexspb

fbshipit-source-id: b5b9a1536fb516e90b5e4b615403086943cfbe93
2017-12-03 10:32:08 -08:00
..
mobile_exporter.py Re-license to Apache 2017-09-28 16:22:00 -07:00
mobile_exporter_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
predictor_exporter.py Allow specifying device to prepare_prediction_net() 2017-12-03 10:32:08 -08:00
predictor_exporter_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
predictor_py_utils.py Add a EmptyDeviceScope (i.e. allow setting CurrentDeviceScope() to None) 2017-11-02 11:25:48 -07:00
predictor_test.py Expose Predictor::run_map to Python 2017-10-18 19:32:56 -07:00
serde.py Re-license to Apache 2017-09-28 16:22:00 -07:00