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
mobile_exporter.py Export uint8 tensors as byte string in mobile_exporter and add GivenTensorByteStringToUInt8FillOp (#10385) 2018-08-15 14:26:50 -07:00
mobile_exporter_test.py Allow mobile exporter to export string arrays (#10017) 2018-08-01 16:09:58 -07:00
predictor_exporter.py Update from Facebook (#8887) 2018-06-26 14:55:48 -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_py_utils.py
predictor_test.py
serde.py