mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
75d6cbe605
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74387 Make temporary python bindings for flatbuffer to test ScriptModule save / load. (Note: this ignores all push blocking failures!) Test Plan: unittest Reviewed By: iseeyuan Differential Revision: D34968080 fbshipit-source-id: d23b16abda6e4b7ecf6b1198ed6e00908a3db903 (cherry picked from commit 5cbbc390c5f54146a1c469106ab4a6286c754325)
10 lines
517 B
Python
10 lines
517 B
Python
from torch._C import LiteScriptModule, ScriptModule
|
|
|
|
def _load_mobile_module_from_file(filename: str): ...
|
|
def _load_mobile_module_from_bytes(bytes_: bytes): ...
|
|
def _load_jit_module_from_file(filename: str): ...
|
|
def _load_jit_module_from_bytes(bytes_: bytes): ...
|
|
def _save_mobile_module(m: LiteScriptModule, filename: str): ...
|
|
def _save_jit_module(m: ScriptModule, filename: str): ...
|
|
def _save_mobile_module_to_bytes(m: LiteScriptModule) -> bytes: ...
|
|
def _save_jit_module_to_bytes(m: ScriptModule) -> bytes: ...
|