1 #include <torch/csrc/jit/api/module.h>
2 #include <torch/csrc/jit/serialization/export.h>
3
4 namespace torch::jit {
5
save(std::ostream & out,const ExtraFilesMap & extra_files) const6 void Module::save(std::ostream& out, const ExtraFilesMap& extra_files) const {
7 ExportModule(*this, out, extra_files, false /* bytecode_format */);
8 }
9
save(const std::string & filename,const ExtraFilesMap & extra_files) const10 void Module::save(const std::string& filename, const ExtraFilesMap& extra_files)
11 const {
12 ExportModule(*this, filename, extra_files, false /* bytecode_format */);
13 }
14
_save_for_mobile(std::ostream & out,const ExtraFilesMap & extra_files,bool save_mobile_debug_info,bool use_flatbuffer) const15 void Module::_save_for_mobile(
16 std::ostream& out,
17 const ExtraFilesMap& extra_files,
18 bool save_mobile_debug_info,
19 bool use_flatbuffer) const {
20 ExportModule(
21 *this,
22 out,
23 extra_files,
24 true /* bytecode_format */,
25 save_mobile_debug_info,
26 use_flatbuffer);
27 }
28
_save_for_mobile(const std::string & filename,const ExtraFilesMap & extra_files,bool save_mobile_debug_info,bool use_flatbuffer) const29 void Module::_save_for_mobile(
30 const std::string& filename,
31 const ExtraFilesMap& extra_files,
32 bool save_mobile_debug_info,
33 bool use_flatbuffer) const {
34 ExportModule(
35 *this,
36 filename,
37 extra_files,
38 true /* bytecode_format */,
39 save_mobile_debug_info,
40 use_flatbuffer);
41 }
42
43 } // namespace torch::jit
44