diff options
| author | Peter Goldsborough <peter@goldsborough.me> | 2018-05-09 14:01:19 -0700 |
|---|---|---|
| committer | Edward Z. Yang <ezyang@mit.edu> | 2018-05-09 17:01:19 -0400 |
| commit | 3023dd25f3586d45097500bb5d9187f2c141d44f (patch) | |
| tree | a1e4868d2586e3624ef8964343b01b7067027d8e /caffe | |
| parent | ed111619da10a976e06dfe04ac7dbe3fc3c25ebc (diff) | |
| download | pytorch-3023dd25f3586d45097500bb5d9187f2c141d44f.tar.gz pytorch-3023dd25f3586d45097500bb5d9187f2c141d44f.tar.bz2 pytorch-3023dd25f3586d45097500bb5d9187f2c141d44f.zip | |
Use set_type to implement type conversions in C++ API (#7408)
* Use set_type to implement .cuda() in C++ API
* Change C++ module parameter types in place
* Fix bug where batchnorm state was not moved to CUDA
Diffstat (limited to 'caffe')
0 files changed, 0 insertions, 0 deletions