pytorch
c0a7c1d0 - Expose _export_data from C++ to Python (#79207)

Commit
2 years ago
Expose _export_data from C++ to Python (#79207) Summary: https://www.internalfb.com/code/fbsource/[477a5768452957f87e56044169de47f051197567]/fbcode/caffe2/torch/csrc/jit/mobile/train/export_data.cpp export_data is used to serialize data. I binded this method to Python with PyBind11 Test Plan: Wrote a file pybind_check.py which checks if the binding works. Then, tried to read the produced data file from C++ with "torch::jit::_load_parameters" and checked that content matched. Differential Revision: D37029253 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79207 Approved by: https://github.com/qihqi
Author
Luka Mushkudiani
Committer
Parents
Loading