OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
41d35dc201
pytorch/torch/csrc/jit/serialization
History
Han Qi 41d35dc201 Add ability for a mobile::Module to save as flatbuffer (#67351) 
Summary:
Included functions:

* save_mobile_module -> saves a mobile::Module to flatbuffer
* load_mobile_module_from_file -> loads a flatbuffer into mobile::Module
* parse_mobile_module -> parses from bytes or deserialized flatbuffer
      Module object

Fixes #{issue number}

Pull Request resolved: https://github.com/pytorch/pytorch/pull/67351

Reviewed By: iseeyuan

Differential Revision: D32010095

Pulled By: qihqi

fbshipit-source-id: d763b0557780f7c2661b6485105b045e41a5e8f1
2021-12-01 23:58:15 -08:00
..
callstack_debug_info_serialization.cpp [PyTorch] Adopt IValue::toTupleRef() where obvious (#65505) 2021-11-02 10:22:18 -07:00
callstack_debug_info_serialization.h
export_bytecode.cpp Refactor saving jit::Module to mobile .pt in 2 steps: (#66494) 2021-11-17 12:02:20 -08:00
export_bytecode.h Refactor saving jit::Module to mobile .pt in 2 steps: (#66494) 2021-11-17 12:02:20 -08:00
export_module.cpp Refactor saving jit::Module to mobile .pt in 2 steps: (#66494) 2021-11-17 12:02:20 -08:00
export.cpp [ONNX] Remove the argument use_external_data_format of export() method entirely. (#67080) (#67811) 2021-11-15 17:20:04 -08:00
export.h [ONNX] Update onnx function export with comments and clean up (#66817) (#67803) 2021-11-05 10:35:35 -07:00
flatbuffer_serializer.cpp Add ability for a mobile::Module to save as flatbuffer (#67351) 2021-12-01 23:58:15 -08:00
flatbuffer_serializer.h Add ability for a mobile::Module to save as flatbuffer (#67351) 2021-12-01 23:58:15 -08:00
import_export_constants.h
import_export_functions.h [PyTorch] Add tuple inline storage (#64066) 2021-10-15 12:16:51 -07:00
import_export_helpers.cpp Add SourceView which doesn't own source text as base class of Source (#65309) 2021-10-18 23:17:22 -07:00
import_export_helpers.h Add SourceView which doesn't own source text as base class of Source (#65309) 2021-10-18 23:17:22 -07:00
import_legacy.cpp [PyTorch] Add tuple inline storage (#64066) 2021-10-15 12:16:51 -07:00
import_legacy.h
import_read.cpp
import_read.h
import_source.cpp [jit] Remove graph() call from abstract Function interface. (#65967) 2021-10-27 11:54:26 -07:00
import_source.h [jit] Remove graph() call from abstract Function interface. (#65967) 2021-10-27 11:54:26 -07:00
import.cpp Support Union in TorchScript (#64234) 2021-09-03 06:12:24 -07:00
import.h
mobile_bytecode.fbs Add ability for a mobile::Module to save as flatbuffer (#67351) 2021-12-01 23:58:15 -08:00
onnx.cpp
onnx.h
pickle.cpp
pickle.h
pickler.cpp [jit] Reduce refcounting of Types (#65345) 2021-10-08 09:03:04 -07:00
pickler.h add support for sending cpu sparse tensors over rpc (#62794) 2021-08-29 11:35:00 -07:00
python_print.cpp [jit] Remove graph() call from abstract Function interface. (#65967) 2021-10-27 11:54:26 -07:00
python_print.h
source_range_serialization_impl.h Lock unpickling of source ranges 2021-09-22 20:41:02 -07:00
source_range_serialization.cpp [PyTorch] Adopt IValue::toTupleRef() where obvious (#65505) 2021-11-02 10:22:18 -07:00
source_range_serialization.h Add SourceView which doesn't own source text as base class of Source (#65309) 2021-10-18 23:17:22 -07:00
storage_context.h
type_name_uniquer.cpp
type_name_uniquer.h
unpickler.cpp [PyTorch] Avoid reading file from stream for 0 byte Tensor storage (#67787) 2021-11-09 21:45:05 -08:00
unpickler.h add support for sending cpu sparse tensors over rpc (#62794) 2021-08-29 11:35:00 -07:00