pytorch/test/cpp
Han Qi d3649309e6 [pytorch][PR] Add ability for a mobile::Module to save as flatbuffer (#69306)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/69306

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

Test Plan: unittests

Reviewed By: gmagogsfm

Differential Revision: D32806835

fbshipit-source-id: 71913c6650e225634f878946bd16960d377a7f57
2021-12-09 14:53:31 -08:00
..
api use irange for loops 10 (#69394) 2021-12-09 09:49:34 -08:00
c10d Remove remaining THC code (#69039) 2021-12-08 12:18:08 -08:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit [pytorch][PR] Add ability for a mobile::Module to save as flatbuffer (#69306) 2021-12-09 14:53:31 -08:00
lazy [LT] Upstream LazyView and view ops IR Nodes (#69277) 2021-12-04 08:44:54 -08:00
lite_interpreter_runtime Back out "Revert D30710710: [Pytorch Edge] Support profiling kineto events from external source" (#66421) 2021-10-12 10:55:29 -07:00
monitor torch.monitor - Initial C++ Stats (#68074) 2021-11-18 21:46:23 -08:00
rpc Remove ProcessGroup from TensorPipeAgent initialization (#68128) 2021-11-11 12:28:55 -08:00
tensorexpr [TensorExpr] Allow for 'keepdim' argument in aten::mean in NNC's external call. (#68756) 2021-11-30 00:06:34 -08:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00