| 1234567891011121314151617181920212223242526272829303132333435363738 |
- # mypy: allow-untyped-defs
- # These functions are referenced from the pickle archives produced by
- # ScriptModule.save()
- # These (`build_*`) functions used to be used by `pickler.cpp` to specify
- # the type of the list for certain special types, but now all lists get
- # a type attached and restored via `restore_type_tag` below. The legacy
- # functions should stick around for backwards-compatibility.
- def build_intlist(data):
- return data
- def build_tensorlist(data):
- return data
- def build_doublelist(data):
- return data
- def build_boollist(data):
- return data
- def build_tensor_from_id(data):
- if isinstance(data, int):
- # just the id, can't really do anything
- return data
- def restore_type_tag(value, type_str):
- # The type_ptr is used by the jit unpickler to restore the full static type
- # to container types like list when they are re-loaded, but this doesn't
- # matter for Python, so just return the plain value
- return value
|
