Home
last modified time | relevance | path

Searched refs:ExtraFilesMap (Results 1 – 21 of 21) sorted by relevance

/external/pytorch/torch/csrc/jit/serialization/
Dflatbuffer_serializer.h22 using ExtraFilesMap = std::unordered_map<std::string, std::string>; variable
75 const ExtraFilesMap& extra_files = ExtraFilesMap(),
76 const ExtraFilesMap& jit_sources = ExtraFilesMap(),
81 const ExtraFilesMap& extra_files = ExtraFilesMap(),
82 const ExtraFilesMap& jit_sources = ExtraFilesMap(),
Dexport.h79 const ExtraFilesMap& extra_files,
90 void writeExtraFiles(const Module& module, const ExtraFilesMap& extra_files);
159 const ExtraFilesMap& metadata = ExtraFilesMap(),
167 const ExtraFilesMap& metadata = ExtraFilesMap(),
175 const ExtraFilesMap& metadata = ExtraFilesMap(),
191 using ExportModuleExtraFilesHook = std::function<ExtraFilesMap(const Module&)>;
267 const ExtraFilesMap& extra_files = ExtraFilesMap());
271 const ExtraFilesMap& extra_files = ExtraFilesMap());
275 const ExtraFilesMap& extra_files,
Dimport.h40 ExtraFilesMap& extra_files,
56 ExtraFilesMap& extra_files,
64 ExtraFilesMap& extra_files,
71 ExtraFilesMap& extra_files,
86 ExtraFilesMap& extra_files,
102 ExtraFilesMap& extra_files,
118 ExtraFilesMap& extra_files,
123 const ExtraFilesMap& source,
130 ExtraFilesMap& extra_files,
135 ExtraFilesMap& extra_files,
[all …]
Dimport.cpp154 ExtraFilesMap& extra_files,
250 ExtraFilesMap& extra_files, in deserialize()
310 ExtraFilesMap extra_files; in import_ir_module()
320 ExtraFilesMap& extra_files,
326 ExtraFilesMap& extra_files, in parse_and_initialize_jit_module()
329 ExtraFilesMap jit_files; in parse_and_initialize_jit_module()
345 ExtraFilesMap& extra_files, in load_jit_module_from_file()
354 ExtraFilesMap& extra_files, in load_jit_module_from_stream()
365 ExtraFilesMap& extra_files, in import_ir_module()
395 ExtraFilesMap extra_files; in import_ir_module()
[all …]
Dflatbuffer_serializer.cpp81 const ExtraFilesMap& extra_files = ExtraFilesMap(),
82 const ExtraFilesMap& jit_sources = ExtraFilesMap(),
153 const ExtraFilesMap& extra_files);
351 const ExtraFilesMap& extra_files) { in storeExtraFilesAndGetOffset()
369 const ExtraFilesMap& extra_files, in serializeModule()
370 const ExtraFilesMap& jit_sources, in serializeModule()
798 const ExtraFilesMap& extra_files, in save_mobile_module()
799 const ExtraFilesMap& jit_sources, in save_mobile_module()
830 const ExtraFilesMap& extra_files, in save_mobile_module_to_bytes()
831 const ExtraFilesMap& jit_sources, in save_mobile_module_to_bytes()
Dexport_module.cpp450 const ExtraFilesMap& extra_files, in serialize()
577 const ExtraFilesMap& extra_files) { in writeExtraFiles()
585 ExtraFilesMap hook_files = hook(module); in writeExtraFiles()
833 const ExtraFilesMap& extra_files, in ExportModule()
854 const ExtraFilesMap& extra_files, in ExportModule()
889 const ExtraFilesMap& extra_files) { in save_jit_module()
900 const ExtraFilesMap& extra_files) { in save_jit_module_to_bytes()
901 ExtraFilesMap jitfiles; in save_jit_module_to_bytes()
911 const ExtraFilesMap& extra_files, in save_jit_module_to_write_func()
922 const ExtraFilesMap& extra_files, in ExportModule()
Dpython_print.h52 ExtraFilesMap* jit_sources, // output
Dpython_print.cpp1739 ExtraFilesMap* jit_sources, // output in jitModuleToPythonCodeAndConstants()
/external/pytorch/torch/csrc/jit/mobile/
Dflatbuffer_loader.h30 using ExtraFilesMap = std::unordered_map<std::string, std::string>; variable
61 ExtraFilesMap* extra_files = nullptr,
77 ExtraFilesMap* extra_files = nullptr);
87 ExtraFilesMap& jit_sources,
90 ExtraFilesMap* extra_files = nullptr);
103 ExtraFilesMap* extra_files = nullptr);
117 ExtraFilesMap* extra_files = nullptr);
128 ExtraFilesMap* extra_files,
Dimport.h12 using ExtraFilesMap = std::unordered_map<std::string, std::string>; variable
23 ExtraFilesMap& extra_file,
29 ExtraFilesMap& extra_files);
34 ExtraFilesMap& extra_files,
40 ExtraFilesMap& extra_files,
70 ExtraFilesMap& extra_files);
Dimport.cpp196 ExtraFilesMap& extra_files);
199 ExtraFilesMap& extra_files);
418 ExtraFilesMap& extra_files) { in deserialize_only_extra()
432 ExtraFilesMap& extra_files) { in deserialize()
504 ExtraFilesMap& extra_files, in _load_for_mobile_impl()
577 ExtraFilesMap& extra_files, in _load_mobile_from_bytes()
603 ExtraFilesMap extra_files; in _load_for_mobile()
610 ExtraFilesMap extra_files; in _load_for_mobile()
617 ExtraFilesMap extra_files; in _load_for_mobile()
624 ExtraFilesMap& extra_files, in _load_for_mobile()
[all …]
Dflatbuffer_loader.cpp90 ExtraFilesMap* jit_sources,
256 ExtraFilesMap* extra_files) { in parseExtraFilesFromVector()
265 ExtraFilesMap& extra_files) { in parseExtraFiles()
729 ExtraFilesMap* jit_sources, in extractJitSourceAndConstants()
762 ExtraFilesMap* extra_files, in parse_and_initialize_mobile_module()
791 ExtraFilesMap* extra_files) { in parse_and_initialize_mobile_module()
805 ExtraFilesMap& jit_sources, in parse_and_initialize_mobile_module_for_jit()
808 ExtraFilesMap* extra_files) { in parse_and_initialize_mobile_module_for_jit()
835 ExtraFilesMap* extra_files) { in load_mobile_module_from_file()
895 ExtraFilesMap* extra_files) { in load_mobile_module_from_stream_with_copy()
/external/pytorch/torch/csrc/jit/api/
Dmodule_save.cpp6 void Module::save(std::ostream& out, const ExtraFilesMap& extra_files) const { in save()
10 void Module::save(const std::string& filename, const ExtraFilesMap& extra_files) in save()
17 const ExtraFilesMap& extra_files, in _save_for_mobile()
31 const ExtraFilesMap& extra_files, in _save_for_mobile()
Dmodule.h41 using ExtraFilesMap = std::unordered_map<std::string, std::string>; variable
221 const ExtraFilesMap& extra_files = ExtraFilesMap()) const;
225 const ExtraFilesMap& extra_files = ExtraFilesMap()) const;
229 const ExtraFilesMap& extra_files = ExtraFilesMap(),
235 const ExtraFilesMap& extra_files = ExtraFilesMap(),
682 using ExtraFilesMap = ::torch::jit::ExtraFilesMap; variable
/external/pytorch/torch/csrc/jit/python/
Dscript_init.cpp688 ExtraFilesMap extra_files_from_python(const py::dict& pydict) { in extra_files_from_python()
689 ExtraFilesMap r; in extra_files_from_python()
696 void extra_files_to_python(const ExtraFilesMap& m, const py::dict& pydict) { in extra_files_to_python()
1050 const ExtraFilesMap& _extra_files = ExtraFilesMap()) { in initJitScriptBindings()
1054 py::arg("_extra_files") = ExtraFilesMap()) in initJitScriptBindings()
1057 [](Module& m, const ExtraFilesMap& _extra_files = ExtraFilesMap()) { in initJitScriptBindings()
1062 py::arg("_extra_files") = ExtraFilesMap()) in initJitScriptBindings()
1067 const ExtraFilesMap& _extra_files = ExtraFilesMap(), in initJitScriptBindings()
1077 py::arg("_extra_files") = ExtraFilesMap(), in initJitScriptBindings()
1083 const ExtraFilesMap& _extra_files = ExtraFilesMap(), in initJitScriptBindings()
[all …]
/external/pytorch/test/cpp/jit/
Dtest_save_load.cpp26 ExtraFilesMap files; in roundtripThroughMobile()
66 SetExportModuleExtraFilesHook([](const Module&) -> ExtraFilesMap { in TEST()
85 ExtraFilesMap extra; in TEST()
91 ExtraFilesMap extra; in TEST()
103 SetExportModuleExtraFilesHook([](const Module&) -> ExtraFilesMap { in TEST()
107 ExtraFilesMap extra; in TEST()
114 ExtraFilesMap extra; in TEST()
Dtest_backend.cpp392 lm._save_for_mobile(ss, ExtraFilesMap(), true); in TEST()
441 lm._save_for_mobile(ss, ExtraFilesMap(), true); in TEST()
502 lm._save_for_mobile(ss, ExtraFilesMap(), true); in TEST()
570 lm._save_for_mobile(ss, ExtraFilesMap(), true); in TEST()
694 c._save_for_mobile(ss, ExtraFilesMap(), true); in TEST()
789 c._save_for_mobile(ss, ExtraFilesMap(), true); in TEST()
Dtest_flatbuffer.cpp1313 ExtraFilesMap extra_files; in jitModuleFromBuffer()
Dtest_lite_interpreter.cpp1454 c._save_for_mobile(ss, ExtraFilesMap(), true); in TEST()
/external/pytorch/torch/csrc/jit/mobile/compatibility/
Dbackport_manager.cpp398 ExtraFilesMap extra_files; in backport_v6_to_v5()
466 ExtraFilesMap extra_files; in backport_v7_to_v6()
500 ExtraFilesMap extra_files; in backport_v9_to_v8()
532 ExtraFilesMap extra_files; in backport_v8_to_v7()
/external/pytorch/docs/source/notes/
Dlarge_scale_deployments.rst124 ExtraFilesMap files;