pytorch/torch/csrc/jit/mobile
Martin Yuan a1fef453b6 Support extra files in _load_for_mobile (#47425)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47425

Extra files can be exported in lite interpreter model, but it could not be loaded. This PR is to add the capability to load extra files from lite interpreter model. Because extra_files is a default argument, it should not affect the existing usage of _load_for_mobile. It's a simple assembly or a generic unordered_map. No additional dependency should be introduced and the size overhead should be small (to be tested).

Test Plan: Imported from OSS

Reviewed By: kwanmacher

Differential Revision: D24770266

Pulled By: iseeyuan

fbshipit-source-id: 7e8bd301ce734dbbf36ae56c9decb045aeb801ce
2020-11-06 20:26:54 -08:00
..
optim Implement a light SGD optimizer (#42137) 2020-08-03 17:27:53 -07:00
export_data.cpp add _save_parameters to serialize map (#43163) 2020-08-18 14:58:04 -07:00
export_data.h add _save_parameters to serialize map (#43163) 2020-08-18 14:58:04 -07:00
function.cpp [pytorch] Make mobile find_method return an optional (#43965) 2020-09-03 14:46:18 -07:00
function.h [pytorch] Make mobile find_method return an optional (#43965) 2020-09-03 14:46:18 -07:00
import_data.cpp [PyTorch][QPL] Add instance_key into MOBILE_MODULE_LOAD_STATS logging. (#45518) 2020-09-30 23:31:35 -07:00
import_data.h add _save_parameters to serialize map (#43163) 2020-08-18 14:58:04 -07:00
import.cpp Support extra files in _load_for_mobile (#47425) 2020-11-06 20:26:54 -08:00
import.h Support extra files in _load_for_mobile (#47425) 2020-11-06 20:26:54 -08:00
interpreter.cpp add overload name for str cmp (#39607) 2020-06-06 23:21:35 -07:00
interpreter.h [JIT] clang-format JIT code (#35115) 2020-03-26 11:24:51 -07:00
method.h [pytorch] Add logging to mobile Method run (#44234) 2020-09-10 15:26:33 -07:00
module.cpp [PyTorch][QPL] Add instance_key into MOBILE_MODULE_STATS logging. (#45517) 2020-09-30 20:13:33 -07:00
module.h For torch::jit::module's attr method to moble::module (#47059) 2020-11-02 16:38:12 -08:00
observer.cpp Add module level qpl logging. (#30906) 2019-12-06 15:52:26 -08:00
observer.h [PyTorch][QPL] Add instance_key into MOBILE_MODULE_LOAD_STATS logging. (#45518) 2020-09-30 23:31:35 -07:00
sequential.cpp Add lite SequentialSampler to torch mobile (#43299) 2020-08-24 09:45:24 -07:00
sequential.h Add lite SequentialSampler to torch mobile (#43299) 2020-08-24 09:45:24 -07:00
type_parser.cpp [WIP] Bind in CellParams for RNN (#35787) 2020-04-28 21:47:06 -07:00
type_parser.h Move special ops into interpreter (#32889) 2020-02-18 15:07:01 -08:00