pytorch/caffe2/python/predictor
Junjie Bai ff608a9ff3 Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12232

Original commit changeset: fca91fea58b7

This adds proper modifications to the DeviceType <->DeviceOption conversion code added in D10033396

Reviewed By: jerryzh168

Differential Revision: D10132473

fbshipit-source-id: 801ef777e2950982cb47b48051b1471a0a91e64b
2018-10-01 21:54:52 -07:00
..
__init__.py Experimental support for setup.py develop mode install 2018-02-12 23:36:18 -08:00
mobile_exporter_test.py Allow mobile exporter to export string arrays (#10017) 2018-08-01 16:09:58 -07:00
mobile_exporter.py Export uint8 tensors as byte string in mobile_exporter and add GivenTensorByteStringToUInt8FillOp (#10385) 2018-08-15 14:26:50 -07:00
predictor_exporter_test.py Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
predictor_exporter.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
predictor_py_utils.py Update from facebook (#8384) 2018-06-13 13:10:45 -07:00
predictor_test.py Expose Predictor::run_map to Python 2017-10-18 19:32:56 -07:00
serde.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00