pytorch/torch/csrc/jit/frontend
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
..
builtin_functions.cpp Throws runtime error when torch.full would infer a float dtype from a bool or integral fill value (#40364) 2020-06-23 23:27:22 -07:00
builtin_functions.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
canonicalize_modified_loop.cpp [JIT] clang-format JIT code (#35115) 2020-03-26 11:24:51 -07:00
canonicalize_modified_loop.h
code_template.h
concrete_module_type.cpp [JIT] Add API for ignoring arbitrary module attributes (#45262) 2020-10-06 18:02:06 -07:00
concrete_module_type.h [JIT] Add API for ignoring arbitrary module attributes (#45262) 2020-10-06 18:02:06 -07:00
convert_to_ssa.cpp Various cleanups to ir_emitter and friends (#46686) 2020-10-28 16:28:05 -07:00
convert_to_ssa.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
edit_distance.cpp [JIT] clang-format JIT code (#35115) 2020-03-26 11:24:51 -07:00
edit_distance.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
error_report.cpp [JIT] recursively compile class types (#38050) 2020-05-12 17:16:28 -07:00
error_report.h [JIT] recursively compile class types (#38050) 2020-05-12 17:16:28 -07:00
exit_transforms.cpp Various cleanups to ir_emitter and friends (#46686) 2020-10-28 16:28:05 -07:00
exit_transforms.h
function_schema_parser.cpp add support for optional int list with scalar fill (#43262) 2020-08-21 18:24:36 -07:00
function_schema_parser.h [RELAND] New operator registration API (#35061) (#35629) 2020-03-29 19:48:29 -07:00
inline_loop_condition.cpp [JIT] clang-format JIT code (#35115) 2020-03-26 11:24:51 -07:00
inline_loop_condition.h
ir_emitter.cpp Support extra files in _load_for_mobile (#47425) 2020-11-06 20:26:54 -08:00
ir_emitter.h [JIT] clang-format JIT code (#35115) 2020-03-26 11:24:51 -07:00
lexer.cpp Support left and right shift operators in JIT (#34563) 2020-03-13 13:00:33 -07:00
lexer.h Add Ellipsis constant to the list of recognized tokens (#44959) 2020-09-22 09:05:25 -07:00
mini_environment.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
name_mangler.cpp [jit] factor mangler out (#35716) 2020-03-31 22:27:39 -07:00
name_mangler.h [jit] factor mangler out (#35716) 2020-03-31 22:27:39 -07:00
parse_string_literal.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
parser_constants.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
parser.cpp Add Ellipsis constant to the list of recognized tokens (#44959) 2020-09-22 09:05:25 -07:00
parser.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
resolver.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
schema_matching.cpp Various cleanups to ir_emitter and friends (#46686) 2020-10-28 16:28:05 -07:00
schema_matching.h Various cleanups to ir_emitter and friends (#46686) 2020-10-28 16:28:05 -07:00
schema_type_parser.cpp Define the record_stream method in native_functions.yaml (#44301) 2020-10-13 09:15:22 -07:00
schema_type_parser.h [JIT] Teach IRPrinter and IRParser to handle 'requires_grad' and 'device' as a part of type info. (#41507) 2020-07-17 10:27:04 -07:00
script_type_parser.cpp [JIT] Add property support to TorchScript classes (#42389) 2020-08-14 12:56:57 -07:00
script_type_parser.h [JIT] use python type resolver for all types (#39880) 2020-06-11 18:36:31 -07:00
source_range.cpp Error printing extension support for multiline errors (#43807) 2020-09-01 10:02:43 -07:00
source_range.h Add CHECK-SOURCE-HIGHLIGHTED to file check utils. (#39692) 2020-06-11 23:47:07 -07:00
string_to_type.cpp Define the record_stream method in native_functions.yaml (#44301) 2020-10-13 09:15:22 -07:00
strtod.cpp [JIT] clang-format JIT code (#35115) 2020-03-26 11:24:51 -07:00
strtod.h [JIT] clang-format JIT code (#35115) 2020-03-26 11:24:51 -07:00
sugared_value.cpp [JIT] Enable ModuleDict non-literal indexing (#45716) 2020-10-31 21:36:23 -07:00
sugared_value.h [JIT] Enable ModuleDict non-literal indexing (#45716) 2020-10-31 21:36:23 -07:00
tracer.cpp Define the record_stream method in native_functions.yaml (#44301) 2020-10-13 09:15:22 -07:00
tracer.h Define the record_stream method in native_functions.yaml (#44301) 2020-10-13 09:15:22 -07:00
tree_views.h [JIT] Add property support to TorchScript classes (#42389) 2020-08-14 12:56:57 -07:00
tree.h Fix grammar and spelling errors (#46713) 2020-10-23 01:31:17 -07:00
versioned_symbols.cpp Adds dynamic versioning pattern (#40279) 2020-06-24 12:52:50 -07:00
versioned_symbols.h Adds dynamic versioning pattern (#40279) 2020-06-24 12:52:50 -07:00