FM / torch

/usr/local/lib64/python3.6/site-packages/torch/

/usr/local/lib64/python3.6/site-packages/torch

Name	Size	Mode	Actions
ao/	-	0755	rm
autograd/	-	0755	rm
backends/	-	0755	rm
bin/	-	0755	rm
contrib/	-	0755	rm
cpu/	-	0755	rm
cuda/	-	0755	rm
distributed/	-	0755	rm
distributions/	-	0755	rm
fft/	-	0755	rm
for_onnx/	-	0755	rm
futures/	-	0755	rm
fx/	-	0755	rm
include/	-	0755	rm
jit/	-	0755	rm
lib/	-	0755	rm
linalg/	-	0755	rm
multiprocessing/	-	0755	rm
nn/	-	0755	rm
onnx/	-	0755	rm
optim/	-	0755	rm
package/	-	0755	rm
profiler/	-	0755	rm
quantization/	-	0755	rm
share/	-	0755	rm
sparse/	-	0755	rm
special/	-	0755	rm
testing/	-	0755	rm
utils/	-	0755	rm
_C/	-	0755	rm
__pycache__/	-	0755	rm
autocast_mode.py	9663	0644	edit dl rm
functional.py	71257	0644	edit dl rm
hub.py	23371	0644	edit dl rm
overrides.py	84146	0644	edit dl rm
py.typed	0	0644	edit dl rm
quasirandom.py	7489	0644	edit dl rm
random.py	4828	0644	edit dl rm
serialization.py	36544	0644	edit dl rm
storage.py	5762	0644	edit dl rm
torch_version.py	3468	0644	edit dl rm
types.py	1552	0644	edit dl rm
version.py	125	0644	edit dl rm
_appdirs.py	26245	0644	edit dl rm
_C.cpython-36m-x86_64-linux-gnu.so	29296	0755	edit dl rm
_classes.py	1717	0644	edit dl rm
_deploy.py	3100	0644	edit dl rm
_dl.cpython-36m-x86_64-linux-gnu.so	29832	0755	edit dl rm
_jit_internal.py	46910	0644	edit dl rm
_linalg_utils.py	2373	0644	edit dl rm
_lobpcg.py	44076	0644	edit dl rm
_lowrank.py	11031	0644	edit dl rm
_namedtensor_internals.py	5351	0644	edit dl rm
_ops.py	4457	0644	edit dl rm
_python_dispatcher.py	7011	0644	edit dl rm
_six.py	1858	0644	edit dl rm
_sources.py	3882	0644	edit dl rm
_storage_docs.py	1298	0644	edit dl rm
_tensor.py	50854	0644	edit dl rm
_tensor_docs.py	113474	0644	edit dl rm
_tensor_str.py	18091	0644	edit dl rm
_torch_docs.py	364678	0644	edit dl rm
_utils.py	21617	0644	edit dl rm
_utils_internal.py	1687	0644	edit dl rm
_VF.py	656	0644	edit dl rm
_vmap_internals.py	13219	0644	edit dl rm
__config__.py	551	0644	edit dl rm
__future__.py	813	0644	edit dl rm
__init__.py	31199	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/_deploy.py (3100B)

import io
import torch
from torch.package._package_pickler import create_pickler
from torch.package._package_unpickler import PackageUnpickler
from torch.package import sys_importer, OrderedImporter, PackageImporter, Importer
from torch.serialization import _maybe_decode_ascii

def _save_storages(importer, obj):
    serialized_storages = []
    serialized_dtypes = []

importer = importer if isinstance(importer, torch.package.PackageImporter) else None
    importers: Importer
    if importer is not None:
        importers = OrderedImporter(importer, sys_importer)
    else:
        importers = sys_importer

def persistent_id(obj):
        # FIXME: the docs say that persistent_id should only return a string
        # but torch store returns tuples. This works only in the binary protocol
        # see
        # https://docs.python.org/2/library/pickle.html#pickling-and-unpickling-external-objects
        # https://github.com/python/cpython/blob/master/Lib/pickle.py#L527-L537
        if torch.is_storage(obj):
            serialized_storages.append(obj)
            serialized_dtypes.append(obj.dtype)
            return ('storage', len(serialized_storages) - 1)

if hasattr(obj, "__reduce_deploy__"):
            if _serialized_reduces.get(id(obj)) is None:
                _serialized_reduces[id(obj)] = (
                    "reduce_deploy",
                    id(obj),
                    *obj.__reduce_deploy__(importers),
                )
            return _serialized_reduces[id(obj)]

return None

# Write the pickle data for `obj`
    data_buf = io.BytesIO()
    pickler = create_pickler(data_buf, importers)
    pickler.persistent_id = persistent_id
    pickler.dump(obj)
    data_value = data_buf.getvalue()
    return data_value, serialized_storages, serialized_dtypes, importer.zip_reader if importer else None

def _load_storages(id, zip_reader, obj_bytes, serialized_storages):

def persistent_load(saved_id):
        assert isinstance(saved_id, tuple)
        typename = _maybe_decode_ascii(saved_id[0])
        data = saved_id[1:]

if typename == 'storage':
            return serialized_storages[data[0]]

if typename == 'reduce_deploy':
            reduce_id, func, args = data
            if reduce_id not in _loaded_reduces:
                _loaded_reduces[reduce_id] = func(_raw_packages[zip_reader], *args)
            return _loaded_reduces[reduce_id]

return None

importer: Importer
    if zip_reader is not None:
        importer = OrderedImporter(_get_package(zip_reader), sys_importer)
    else:
        importer = sys_importer

unpickler = PackageUnpickler(importer, io.BytesIO(obj_bytes))
    unpickler.persistent_load = persistent_load
    result = _deploy_objects[id] = unpickler.load()
    return result

def _get_package(zip_reader):
    if zip_reader not in _raw_packages:
        _raw_packages[zip_reader] = PackageImporter(zip_reader)
    return _raw_packages[zip_reader]

_raw_packages: dict = {}
_deploy_objects: dict = {}
_serialized_reduces: dict = {}
_loaded_reduces: dict = {}

cmd: