| #include <torch/csrc/jit/serialization/pickle.h> | |
| #include <torch/csrc/jit/serialization/pickler.h> | |
| #include <torch/serialize.h> | |
| #include <vector> | |
| namespace torch { | |
| std::vector<char> pickle_save(const at::IValue& ivalue) { | |
| return jit::pickle_save(ivalue); | |
| } | |
| torch::IValue pickle_load(const std::vector<char>& data) { | |
| return jit::pickle_load(data); | |
| } | |
| } // namespace torch |